From 38b3c3c83d811768c81a32e24ed1d75752c99e46 Mon Sep 17 00:00:00 2001
From: Scott Zhu <scottzhu@google.com>
Date: Tue, 8 Dec 2020 16:12:10 -0800
Subject: [PATCH] Update metrics.py with proper import and build deps.
PiperOrigin-RevId: 346432463
Change-Id: I5ededfa14d095b6ed0a15d4401325c69a9a5099a
---
tensorflow/python/keras/BUILD | 1 +
tensorflow/python/keras/metrics.py | 4 ++--
2 files changed, 3 insertions(+), 2 deletions(-)
diff --git a/tensorflow/python/keras/BUILD b/tensorflow/python/keras/BUILD
index 612a3855ff9..4e8fed1b4e3 100755
--- a/tensorflow/python/keras/BUILD
+++ b/tensorflow/python/keras/BUILD
@@ -236,6 +236,7 @@ py_library(
],
srcs_version = "PY2AND3",
deps = [
+ ":activations",
":backend",
":losses",
"//tensorflow/python:array_ops",
diff --git a/tensorflow/python/keras/metrics.py b/tensorflow/python/keras/metrics.py
index f05fb910a72..5b7019793e6 100644
--- a/tensorflow/python/keras/metrics.py
+++ b/tensorflow/python/keras/metrics.py
@@ -37,8 +37,8 @@ from tensorflow.python.framework import dtypes
from tensorflow.python.framework import ops
from tensorflow.python.framework import tensor_shape
from tensorflow.python.framework import tensor_spec
+from tensorflow.python.keras import activations
from tensorflow.python.keras import backend as K
-from tensorflow.python.keras.activations import sigmoid
from tensorflow.python.keras.engine import base_layer
from tensorflow.python.keras.engine import base_layer_utils
from tensorflow.python.keras.engine import keras_tensor
@@ -2134,7 +2134,7 @@ class AUC(Metric):
label_weights = None if self.multi_label else self.label_weights
if self._from_logits:
- y_pred = sigmoid(y_pred)
+ y_pred = activations.sigmoid(y_pred)
with ops.control_dependencies(deps):
return metrics_utils.update_confusion_matrix_variables(