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 -
Method Summary
Modifier and TypeMethodDescriptionvoidaddMemoryUsage(long value) voidensureCapacity(long size) intenumerateLabel(String label) longio.airlift.slice.SlicevoidsetGroupId(long groupId) voidsetParameters(io.airlift.slice.Slice parameters) Methods inherited from class java.lang.Object
clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, waitMethods inherited from interface io.trino.spi.function.AccumulatorState
copy
-
Constructor Details
-
GroupedLearnState
public GroupedLearnState()
-
-
Method Details
-
setGroupId
public void setGroupId(long groupId) - Specified by:
setGroupIdin interfaceGroupedAccumulatorState
-
ensureCapacity
public void ensureCapacity(long size) - Specified by:
ensureCapacityin interfaceGroupedAccumulatorState
-
getEstimatedSize
public long getEstimatedSize()- Specified by:
getEstimatedSizein interfaceAccumulatorState
-
getLabelEnumeration
- Specified by:
getLabelEnumerationin interfaceLearnState
-
enumerateLabel
- Specified by:
enumerateLabelin interfaceLearnState
-
getLabels
- Specified by:
getLabelsin interfaceLearnState
-
getFeatureVectors
- Specified by:
getFeatureVectorsin interfaceLearnState
-
getParameters
public io.airlift.slice.Slice getParameters()- Specified by:
getParametersin interfaceLearnState
-
setParameters
public void setParameters(io.airlift.slice.Slice parameters) - Specified by:
setParametersin interfaceLearnState
-
addMemoryUsage
public void addMemoryUsage(long value) - Specified by:
addMemoryUsagein interfaceLearnState
-