From 3b0cdd878ce8d1e1120acd99b8e77401e06a65bb Mon Sep 17 00:00:00 2001 From: Haiping Chen Date: Sun, 26 Feb 2023 12:29:57 -0600 Subject: [PATCH] Fix double data type issue in hamming_loss_fn. --- src/TensorFlowNET.Keras/Metrics/metrics_utils.cs | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/TensorFlowNET.Keras/Metrics/metrics_utils.cs b/src/TensorFlowNET.Keras/Metrics/metrics_utils.cs index 69cc789e..269bb1fb 100644 --- a/src/TensorFlowNET.Keras/Metrics/metrics_utils.cs +++ b/src/TensorFlowNET.Keras/Metrics/metrics_utils.cs @@ -31,7 +31,7 @@ public class metrics_utils threshold = tf.reduce_max(y_pred, axis: -1, keepdims: true); // make sure [0, 0, 0] doesn't become [1, 1, 1] // Use abs(x) > eps, instead of x != 0 to check for zero - y_pred = tf.logical_and(y_pred >= threshold, tf.abs(y_pred) > 1e-12); + y_pred = tf.logical_and(y_pred >= threshold, tf.abs(y_pred) > 1e-12f); } else {