diff --git a/src/TensorFlowNET.Keras/Engine/Model.Train.cs b/src/TensorFlowNET.Keras/Engine/Model.Train.cs index 0090b69e..0151d543 100644 --- a/src/TensorFlowNET.Keras/Engine/Model.Train.cs +++ b/src/TensorFlowNET.Keras/Engine/Model.Train.cs @@ -39,7 +39,15 @@ namespace Tensorflow.Keras.Engine compiled_metrics.update_state(y, y_pred); var dict = new Dictionary(); - metrics.ToList().ForEach(x => dict[x.Name] = (float)x.result()); + metrics.ToList().ForEach(x => + { + var r = x.result(); + if (r.ndim > 0) + { + r = tf.reduce_mean(r); + } + dict[x.Name] = (float)r; + }); return dict; }