Package org.tribuo.sequence
Interface SequenceTrainer<T extends Output<T>>
- All Superinterfaces:
com.oracle.labs.mlrg.olcut.config.Configurable
,com.oracle.labs.mlrg.olcut.provenance.Provenancable<TrainerProvenance>
- All Known Implementing Classes:
CRFTrainer
,HashingSequenceTrainer
,IndependentSequenceTrainer
,TensorFlowSequenceTrainer
,ViterbiTrainer
public interface SequenceTrainer<T extends Output<T>>
extends com.oracle.labs.mlrg.olcut.config.Configurable, com.oracle.labs.mlrg.olcut.provenance.Provenancable<TrainerProvenance>
An interface for things that can train sequence prediction models.
-
Method Summary
Modifier and TypeMethodDescriptionint
Returns the number of times the train method has been invoked.default SequenceModel<T>
train
(SequenceDataset<T> examples) Trains a sequence prediction model using the examples in the given data set.train
(SequenceDataset<T> examples, Map<String, com.oracle.labs.mlrg.olcut.provenance.Provenance> runProvenance) Trains a sequence prediction model using the examples in the given data set.Methods inherited from interface com.oracle.labs.mlrg.olcut.config.Configurable
postConfig
Methods inherited from interface com.oracle.labs.mlrg.olcut.provenance.Provenancable
getProvenance
-
Method Details
-
train
Trains a sequence prediction model using the examples in the given data set.- Parameters:
examples
- the data set containing the examples.- Returns:
- a prediction model that can be used to predict values for new examples.
-
train
SequenceModel<T> train(SequenceDataset<T> examples, Map<String, com.oracle.labs.mlrg.olcut.provenance.Provenance> runProvenance) Trains a sequence prediction model using the examples in the given data set.- Parameters:
examples
- the data set containing the examples.runProvenance
- Training run specific provenance (e.g., fold number).- Returns:
- a predictive model that can be used to generate predictions for new examples.
-
getInvocationCount
int getInvocationCount()Returns the number of times the train method has been invoked.- Returns:
- The number of times train has been invoked.
-