Package io.trino.plugin.ml

Class AbstractSvmModel

java.lang.Object

io.trino.plugin.ml.AbstractSvmModel

All Implemented Interfaces:

Model

Direct Known Subclasses:

SvmClassifier, SvmRegressor

public abstract class AbstractSvmModel
extends Object
implements Model

Field Summary

Fields

Modifier and Type	Field	Description
protected libsvm.svm_model	model
protected libsvm.svm_parameter	params

Constructor Summary

Constructors

Modifier	Constructor	Description
protected	AbstractSvmModel(libsvm.svm_model model)
protected	AbstractSvmModel(libsvm.svm_parameter params)

Method Summary

Modifier and Type	Method	Description
protected abstract int	getLibsvmType()
byte[]	getSerializedData()
protected static libsvm.svm_node[]	toSvmNodes(FeatureVector features)
void	train(Dataset dataset)

Methods inherited from class java.lang.Object

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

Methods inherited from interface io.trino.plugin.ml.Model

getType

Field Details

model

protected libsvm.svm_model model

params

protected libsvm.svm_parameter params

Constructor Details

AbstractSvmModel

protected AbstractSvmModel(libsvm.svm_parameter params)

AbstractSvmModel

protected AbstractSvmModel(libsvm.svm_model model)

Method Details

getSerializedData

public byte[] getSerializedData()

Specified by:

getSerializedData in interface Model

train

public void train(Dataset dataset)

Specified by:

train in interface Model

getLibsvmType

protected abstract int getLibsvmType()

toSvmNodes

protected static libsvm.svm_node[] toSvmNodes(FeatureVector features)