Package io.trino.plugin.ml

Class LearnStateFactory.SingleLearnState

java.lang.Object

io.trino.plugin.ml.LearnStateFactory.SingleLearnState

All Implemented Interfaces:

LearnState, AccumulatorState

Enclosing class:

LearnStateFactory

public static class LearnStateFactory.SingleLearnState
extends Object
implements LearnState

Constructor Summary

Constructors

Constructor	Description
SingleLearnState()

Method Summary

Modifier and Type	Method	Description
void	addMemoryUsage(long value)
int	enumerateLabel(String label)
long	getEstimatedSize()
List<FeatureVector>	getFeatureVectors()
com.google.common.collect.BiMap<String,Integer>	getLabelEnumeration()
List<Double>	getLabels()
io.airlift.slice.Slice	getParameters()
void	setParameters(io.airlift.slice.Slice parameters)

Methods inherited from class java.lang.Object

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

Methods inherited from interface io.trino.spi.function.AccumulatorState

copy

Constructor Details

SingleLearnState

public SingleLearnState()

Method Details

getEstimatedSize

public long getEstimatedSize()

Specified by:

getEstimatedSize in interface AccumulatorState

getLabelEnumeration

public com.google.common.collect.BiMap<String,Integer> getLabelEnumeration()

Specified by:

getLabelEnumeration in interface LearnState

enumerateLabel

public int enumerateLabel(String label)

Specified by:

enumerateLabel in interface LearnState

getLabels

public List<Double> getLabels()

Specified by:

getLabels in interface LearnState

getFeatureVectors

public List<FeatureVector> getFeatureVectors()

Specified by:

getFeatureVectors in interface LearnState

getParameters

public io.airlift.slice.Slice getParameters()

Specified by:

getParameters in interface LearnState

setParameters

public void setParameters(io.airlift.slice.Slice parameters)

Specified by:

setParameters in interface LearnState

addMemoryUsage

public void addMemoryUsage(long value)

Specified by:

addMemoryUsage in interface LearnState