Compare commits

...

2 Commits

Author SHA1 Message Date
bobby
c4f4089abd
Merge 53eb6940f7 into fb54d8b549 2023-05-22 17:41:28 -05:00
bobby
53eb6940f7
No need for OrderedDict, default dictionary has been in order for a while
https://realpython.com/python-ordereddict/ for more info or look at Python changelings.
2023-03-31 19:21:21 -04:00

View File

@ -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']