From efdf172669d4202f9059677ba077f36c42e31561 Mon Sep 17 00:00:00 2001 From: Pedro Spinosa Date: Wed, 10 Jul 2024 17:28:56 -0300 Subject: [PATCH] Lint fix --- src/fklearn/validation/evaluators.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/src/fklearn/validation/evaluators.py b/src/fklearn/validation/evaluators.py index b05d9d32..093dbd08 100644 --- a/src/fklearn/validation/evaluators.py +++ b/src/fklearn/validation/evaluators.py @@ -1065,7 +1065,9 @@ def logistic_coefficient_evaluator(test_data: pd.DataFrame, if eval_name is None: eval_name = "logistic_coefficient_evaluator__" + target_column - score = LogisticRegression(penalty=None, multi_class="ovr").fit(test_data[[prediction_column]], - test_data[target_column]).coef_[0][0] + score = LogisticRegression(penalty=None, multi_class="ovr").fit( + test_data[[prediction_column]], + test_data[target_column] + ).coef_[0][0] return {eval_name: score}