Package io.trino.plugin.ml

Class StringClassifierAdapter

java.lang.Object

io.trino.plugin.ml.StringClassifierAdapter

All Implemented Interfaces:

Classifier<String>, Model

public class StringClassifierAdapter
extends Object
implements Classifier<String>

Constructor Summary

Constructors

Constructor	Description
StringClassifierAdapter(Classifier<Integer> classifier)
StringClassifierAdapter(Classifier<Integer> classifier, Map<Integer,String> labelEnumeration)

Method Summary

Modifier and Type	Method	Description
String	classify(FeatureVector features)
static StringClassifierAdapter	deserialize(byte[] data)
byte[]	getSerializedData()
ModelType	getType()
void	train(Dataset dataset)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Details

StringClassifierAdapter

public StringClassifierAdapter(Classifier<Integer> classifier)

StringClassifierAdapter

public StringClassifierAdapter(Classifier<Integer> classifier,
 Map<Integer,String> labelEnumeration)

Method Details

getType

public ModelType getType()

Specified by:

getType in interface Model

getSerializedData

public byte[] getSerializedData()

Specified by:

getSerializedData in interface Model

deserialize

public static StringClassifierAdapter deserialize(byte[] data)

classify

public String classify(FeatureVector features)

Specified by:

classify in interface Classifier<String>

train

public void train(Dataset dataset)

Specified by:

train in interface Model