Mirrors/the-algorithm
2
0
Fork 0
mirror of https://github.com/twitter/the-algorithm.git synced 2024-12-22 18:21:51 +01:00
Code Issues Packages Projects Releases Wiki Activity

Merge a672d8c520 into fb54d8b549

Browse Source
This commit is contained in:
ibuki420 2023-05-22 17:40:46 -05:00 committed by GitHub
commit 2815c8e415
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/python/twitter/deepbird/projects/timelines/scripts/models/earlybird/metrics.py
View File

@ -52,7 +52,7 @@ def get_multi_binary_class_metric_fn(metrics, classes=None, class_dim=1):
num_labels = shape[class_dim] num_labels = shape[class_dim]
# If we are doing multi-class / multi-label metric, the number of classes / labels must # 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 num_labels is not None, "The multi-metric dimension cannot be None."
assert classes is None or len(classes) == num_labels, ( assert classes is None or len(classes) == num_labels, (
"Number of classes must match the number of labels") "Number of classes must match the number of labels")