LearnState (trino-ml 434 API)

All Superinterfaces:: AccumulatorState

All Known Implementing Classes:: LearnStateFactory.GroupedLearnState, LearnStateFactory.SingleLearnState

public interface LearnState extends AccumulatorState

Method Summary

Modifier and Type

Method

Description

void

addMemoryUsage(long value)

int

enumerateLabel(String label)

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 interface io.trino.spi.function.AccumulatorState
copy, getEstimatedSize

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