Skip to content

Commit

Permalink
Lint fix
Browse files Browse the repository at this point in the history
  • Loading branch information
spinosaphb committed Jul 10, 2024
1 parent eafe0cd commit efdf172
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions src/fklearn/validation/evaluators.py
Original file line number Diff line number Diff line change
Expand Up @@ -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}

0 comments on commit efdf172

Please sign in to comment.