Package io.trino.plugin.ml

Class RegressorFeatureTransformer

java.lang.Object

io.trino.plugin.ml.RegressorFeatureTransformer

All Implemented Interfaces:

Model, Regressor

public class RegressorFeatureTransformer
extends Object
implements Regressor

Constructor Summary

Constructors

Constructor	Description
RegressorFeatureTransformer(Regressor regressor, FeatureTransformation transformation)

Method Summary

Modifier and Type	Method	Description
static RegressorFeatureTransformer	deserialize(byte[] data)
byte[]	getSerializedData()
ModelType	getType()
double	regress(FeatureVector features)
void	train(Dataset dataset)

Methods inherited from class java.lang.Object

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

Constructor Details

RegressorFeatureTransformer

public RegressorFeatureTransformer(Regressor regressor,
 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 RegressorFeatureTransformer deserialize(byte[] data)

regress

public double regress(FeatureVector features)

Specified by:

regress in interface Regressor

train

public void train(Dataset dataset)

Specified by:

train in interface Model