org.tribuo.evaluation.metrics

Class MetricContext<T extends Output<T>>

java.lang.Object

org.tribuo.evaluation.metrics.MetricContext<T>

Direct Known Subclasses:

LabelMetric.Context

public abstract class MetricContext<T extends Output<T>>
extends Object

The context for a metric or set of metrics. At minimum the model used to generate the predictions, and the predictions themselves.

Constructor Summary

Constructors

Modifier	Constructor and Description
protected	MetricContext(Model<T> model, List<Prediction<T>> predictions)
protected	MetricContext(SequenceModel<T> model, List<Prediction<T>> predictions)

Method Summary

Modifier and Type	Method and Description
Model<T>	getModel() Gets the Model used by this context.
List<Prediction<T>>	getPredictions() Gets the predictions used by this context.
SequenceModel<T>	getSequenceModel() Gets the SequenceModel used by this context.

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

MetricContext

protected MetricContext(Model<T> model,
                        List<Prediction<T>> predictions)

MetricContext

protected MetricContext(SequenceModel<T> model,
                        List<Prediction<T>> predictions)

Method Detail

getModel

public Model<T> getModel()

Gets the Model used by this context.

Returns:

The model, or null if this MetricContext operates on a SequenceModel.

getSequenceModel

public SequenceModel<T> getSequenceModel()

Gets the SequenceModel used by this context.

Returns:

The model, or null if this MetricContext operates on a Model.

getPredictions

public List<Prediction<T>> getPredictions()

Gets the predictions used by this context.

Returns:

The predictions.