Compare commits
3 Commits
babc872017
...
2e6bfc8de7
Author | SHA1 | Date |
---|---|---|
ibuki420 | 2e6bfc8de7 | |
ibuki420 | 20889ae89c | |
v420v | a672d8c520 |
|
@ -52,7 +52,7 @@ def get_multi_binary_class_metric_fn(metrics, classes=None, class_dim=1):
|
|||
|
||||
num_labels = shape[class_dim]
|
||||
# If we are doing multi-class / multi-label metric, the number of classes / labels must
|
||||
# be know at graph construction time. This dimension cannot have size None.
|
||||
# be known at graph construction time. This dimension cannot have size None.
|
||||
assert num_labels is not None, "The multi-metric dimension cannot be None."
|
||||
assert classes is None or len(classes) == num_labels, (
|
||||
"Number of classes must match the number of labels")
|
||||
|
|
Loading…
Reference in New Issue