Module mls_lib.model_evaluation.evaluate_accuracy

Accuracy evaluation step.

Classes

class EvaluateAccuracy (features: DataFrame, truth: DataFrame, model: Model)

Accuracy evaluation step.

Expand source code

class EvaluateAccuracy(ModelEvaluationStep):
    """ Accuracy evaluation step. """
    def __init__(self, features : DataFrame, truth : DataFrame, model : Model) -> None:
        super().__init__(
            features = features,
            truth = truth,
            model = model
        )

    def execute(self) -> None:
        model = self._get_input('model')
        x_test = self._get_input('features').get_data()
        y_test = self._get_input('truth').get_data()

        result = model.score(x_test, y_test)

        self._set_output("result", result)

        self._finish_execution()

Ancestors

• ModelEvaluationStep
• Step

Inherited members

• ModelEvaluationStep:
  • execute
  • get_output
  • is_finished
  • is_ready