ModelFox.BinaryClassificationPredictOutput (modelfox v0.8.0) View Source

ModelFox.predict outputs {:binary_classification, BinaryClassificationPredictOutput.t()} when the model's task is binary classification.

class_name

This is the name of the predicted class.

probability

This is the probability the model assigned to the predicted class.

feature_contributions

If computing feature contributions was enabled in the predict options, this value will explain the model's output, showing how much each feature contributed to the output.

Link to this section Summary

Link to this section Types

Specs

t() :: %ModelFox.BinaryClassificationPredictOutput{
  class_name: String.t(),
  feature_contributions: FeatureContributions.t() | nil,
  probability: float()
}