Package io.trino.plugin.ml

Class LearnStateFactory.GroupedLearnState

java.lang.Object

io.trino.plugin.ml.LearnStateFactory.GroupedLearnState

All Implemented Interfaces:

LearnState, AccumulatorState, GroupedAccumulatorState

Enclosing class:

LearnStateFactory

public static class LearnStateFactory.GroupedLearnState
extends Object
implements GroupedAccumulatorState, LearnState

Constructor Summary

Constructors

Constructor	Description
GroupedLearnState()

Method Summary

Modifier and Type	Method	Description
void	addMemoryUsage(long value)
void	ensureCapacity(long size)
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	setGroupId(long groupId)
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

GroupedLearnState

public GroupedLearnState()

Method Details

setGroupId

public void setGroupId(long groupId)

Specified by:

setGroupId in interface GroupedAccumulatorState

ensureCapacity

public void ensureCapacity(long size)

Specified by:

ensureCapacity in interface GroupedAccumulatorState

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