Skip to content

GitLab

Commit fd232c2c authored by Rui Qian's avatar Rui Qian Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 458358467
parent 0f091a60
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 0 deletions
official/projects/yt8m/eval_utils/eval_util.py
View file @ fd232c2c
...@@ -13,6 +13,7 @@ ...@@ -13,6 +13,7 @@
# limitations under the License. # limitations under the License.
"""Provides functions to help with evaluating models.""" """Provides functions to help with evaluating models."""
import logging
import numpy as np import numpy as np
import tensorflow as tf import tensorflow as tf
from official.projects.yt8m.eval_utils import average_precision_calculator as ap_calculator from official.projects.yt8m.eval_utils import average_precision_calculator as ap_calculator
...@@ -57,6 +58,9 @@ def calculate_precision_at_equal_recall_rate(predictions, actuals): ...@@ -57,6 +58,9 @@ def calculate_precision_at_equal_recall_rate(predictions, actuals):
""" """
aggregated_precision = 0.0 aggregated_precision = 0.0
num_videos = actuals.shape[0] num_videos = actuals.shape[0]
if num_videos == 0:
logging.warning("Num_videos is 0, returning 0.0 aggregated_precision.")
return aggregated_precision
for row in np.arange(num_videos): for row in np.arange(num_videos):
num_labels = int(np.sum(actuals[row])) num_labels = int(np.sum(actuals[row]))
top_indices = np.argpartition(predictions[row], -num_labels)[-num_labels:] top_indices = np.argpartition(predictions[row], -num_labels)[-num_labels:]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment