Make except more explicit

This commit is contained in:
Ines Montani 2019-09-18 19:57:08 +02:00
parent 1f648ecb76
commit 2e5ab5b59c

View File

@ -60,7 +60,7 @@ class ROCAUCScore(object):
self.saved_score = _roc_auc_score(self.golds, self.cands)
# catch ValueError: Only one class present in y_true.
# ROC AUC score is not defined in that case.
except:
except ValueError:
self.saved_score = -float("inf")
self.saved_score_at_len = len(self.golds)
return self.saved_score