From 21090633863bf6aacfc20d6ca93531833d0be5e8 Mon Sep 17 00:00:00 2001 From: Haiping Chen Date: Sun, 26 Feb 2023 12:59:00 -0600 Subject: [PATCH] Fix FBetaScore data type issue. --- src/TensorFlowNET.Keras/Metrics/FBetaScore.cs | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/TensorFlowNET.Keras/Metrics/FBetaScore.cs b/src/TensorFlowNET.Keras/Metrics/FBetaScore.cs index 39e3e9af..a40a7caa 100644 --- a/src/TensorFlowNET.Keras/Metrics/FBetaScore.cs +++ b/src/TensorFlowNET.Keras/Metrics/FBetaScore.cs @@ -53,7 +53,7 @@ public class FBetaScore : Metric _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 {