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 Type	Method and Description
int	getInvocationCount() 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.
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.

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 Detail

train

default SequenceModel<T> train(SequenceDataset<T> examples)

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.