Skip to content

Commit

Permalink
fixed causes of linting errors
Browse files Browse the repository at this point in the history
  • Loading branch information
efiegel committed Nov 25, 2020
1 parent 9405ed1 commit 30f3f5d
Showing 1 changed file with 2 additions and 3 deletions.
5 changes: 2 additions & 3 deletions sklearn/metrics/tests/test_score_objects.py
Original file line number Diff line number Diff line change
Expand Up @@ -987,11 +987,10 @@ def test_scorer_general_multiclass():

tree = DecisionTreeClassifier(random_state=0).fit(X_train, y_train)
lr = LogisticRegression(random_state=0).fit(X_train, y_train)

# don't use these scorers since they're not for multiclass
binary_only = ['top_k_accuracy', 'f1', 'roc_auc', 'average_precision',
'precision', 'recall', 'neg_log_loss', 'neg_brier_score',
'jaccard']
'precision', 'recall', 'neg_log_loss', 'neg_brier_score', 'jaccard']

for name in CLF_SCORERS:
if name not in binary_only:
Expand Down

0 comments on commit 30f3f5d

Please sign in to comment.