def eval()

in Project-AutoML/automl/mod/mod_autosklearn.py [0:0]

• 10 lines of code
• 2 McCabe index (conditional complexity)

    def eval(pipeline: Pipeline, test_x, test_y, task="classification"):
        oridinal_encoder = pipeline.steps[0][1]
        classifier = pipeline.steps[1][1]
        test_x = oridinal_encoder.transform(test_x)
        y_pred = classifier.predict(test_x)
        if task == "classification":
            metrics = eval_classification_metrics(test_y, y_pred)
        else:
            metrics = super().eval_automl(automl, test_x, test_y)

        return metrics