mirror of
https://github.com/twitter/the-algorithm.git
synced 2025-02-09 09:03:27 +01:00
Merge 53eb6940f7c55d7d57b90b1f1af92fdc2cc2b660 into 72eda9a24f815f6d566818cbf8518138e29d83e9
This commit is contained in:
commit
315aeace47
@ -1,6 +1,5 @@
|
||||
# checkstyle: noqa
|
||||
import tensorflow.compat.v1 as tf
|
||||
from collections import OrderedDict
|
||||
from .constants import EB_SCORE_IDX
|
||||
from .lolly.data_helpers import get_lolly_scores
|
||||
|
||||
@ -35,7 +34,7 @@ def get_multi_binary_class_metric_fn(metrics, classes=None, class_dim=1):
|
||||
# Added to support per engagement metrics for both TF and Lolly scores.
|
||||
labels = tf.tile(labels, [1, 2])
|
||||
|
||||
eval_metric_ops = OrderedDict()
|
||||
eval_metric_ops = dict()
|
||||
|
||||
preds = graph_output['output']
|
||||
|
||||
|
Loading…
x
Reference in New Issue
Block a user