Skip to content

Commit

Permalink
Update deepchecks/tabular/metric_utils/scorers.py
Browse files Browse the repository at this point in the history
  • Loading branch information
noamzbr committed Feb 22, 2024
1 parent 78435e6 commit 6f22681
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion deepchecks/tabular/metric_utils/scorers.py
Expand Up @@ -261,7 +261,7 @@ def __init__(self, user_model, model_classes, data):
if len(data) > 1:
predictions = predictions.squeeze()
self.predictions = pd.Series(predictions, index=data.index)
self._estimator_type = "classifier"
self._estimator_type = 'classifier'

def predict(self, data: pd.DataFrame) -> np.ndarray:
"""Convert labels to 0/1 if model is a binary classifier."""
Expand Down

0 comments on commit 6f22681

Please sign in to comment.