From 544ff9125b9f1143ee0e0f77ca9bfafb706a65c5 Mon Sep 17 00:00:00 2001
From: wangdapao666 <583087864@qq.com>
Date: Thu, 9 Mar 2023 09:57:45 +0800
Subject: [PATCH] Fix model.evaluate don't have output

---
 src/TensorFlowNET.Core/Keras/Engine/ICallback.cs | 1 +
 src/TensorFlowNET.Keras/Engine/Model.Evaluate.cs | 3 ++-
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/TensorFlowNET.Core/Keras/Engine/ICallback.cs b/src/TensorFlowNET.Core/Keras/Engine/ICallback.cs
index 07b93a75..296c32ac 100644
--- a/src/TensorFlowNET.Core/Keras/Engine/ICallback.cs
+++ b/src/TensorFlowNET.Core/Keras/Engine/ICallback.cs
@@ -13,5 +13,6 @@ public interface ICallback
     void on_predict_batch_end(long end_step, Dictionary<string, Tensors> logs);
     void on_predict_end();
     void on_test_begin();
+    void on_test_batch_begin(long step);
     void on_test_batch_end(long end_step, IEnumerable<(string, Tensor)> logs);
 }
diff --git a/src/TensorFlowNET.Keras/Engine/Model.Evaluate.cs b/src/TensorFlowNET.Keras/Engine/Model.Evaluate.cs
index 8e15428e..5397de47 100644
--- a/src/TensorFlowNET.Keras/Engine/Model.Evaluate.cs
+++ b/src/TensorFlowNET.Keras/Engine/Model.Evaluate.cs
@@ -72,12 +72,13 @@ namespace Tensorflow.Keras.Engine
 
                 foreach (var step in data_handler.steps())
                 {
-                    callbacks.on_train_batch_begin(step);
+                    callbacks.on_test_batch_begin(step);
                     logs = test_function(data_handler, iterator);
                     var end_step = step + data_handler.StepIncrement;
                     callbacks.on_test_batch_end(end_step, logs);
                 }
             }
+            Console.WriteLine();
             GC.Collect();
             GC.WaitForPendingFinalizers();
         }