evalml.objectives.BinaryClassificationObjective

Inheritance diagram of BinaryClassificationObjective
class evalml.objectives.BinaryClassificationObjective[source]

Base class for all binary classification objectives.

problem_type (ProblemTypes): Type of problem this objective is. Set to ProblemTypes.BINARY. can_optimize_threshold (bool): Determines if threshold used by objective can be optimized or not.

Methods

decision_function

Apply a learned threshold to predicted probabilities to get predicted classes.

objective_function

Computes the relative value of the provided predictions compared to the actual labels, according a specified metric

optimize_threshold

Learn a binary classification threshold which optimizes the current objective.

score

Returns a numerical score indicating performance based on the differences between the predicted and actual values.

validate_inputs

Validates the input based on a few simple checks.