Package io.trino.plugin.ml

Class ClassifierFeatureTransformer

java.lang.Object

io.trino.plugin.ml.ClassifierFeatureTransformer

All Implemented Interfaces:

Classifier<Integer>, Model

public class ClassifierFeatureTransformer
extends Object
implements Classifier<Integer>

Constructor Summary

Constructors

Constructor	Description
ClassifierFeatureTransformer(Classifier<Integer> classifier, FeatureTransformation transformation)

Method Summary

Modifier and Type	Method	Description
Integer	classify(FeatureVector features)
static ClassifierFeatureTransformer	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

ClassifierFeatureTransformer

public ClassifierFeatureTransformer(Classifier<Integer> classifier,
 FeatureTransformation transformation)

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 ClassifierFeatureTransformer deserialize(byte[] data)

classify

public Integer classify(FeatureVector features)

Specified by:

classify in interface Classifier<Integer>

train

public void train(Dataset dataset)

Specified by:

train in interface Model