Skip to content

Commit 02b890d

Browse files
YiqinZhaoybubnov
authored andcommitted
Add classes validation (#37)
Raise ValueError for incorrect number of classes in average recall metric.
1 parent 41de08b commit 02b890d

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

keras_metrics/metrics.py

+3
Original file line numberDiff line numberDiff line change
@@ -235,6 +235,9 @@ class average_recall(layer):
235235
def __init__(self, name="average_recall", classes=2, **kwargs):
236236
super(average_recall, self).__init__(name=name, **kwargs)
237237

238+
if classes < 2:
239+
raise ValueError('argument classes must >= 2')
240+
238241
self.classes = classes
239242

240243
self.true = K.zeros(classes, dtype="int32")

0 commit comments

Comments
 (0)