Update NeuralNet_2_0.py (microsoft#522)

Fixed accuracy not returned in Test method.
XerxeLee · Jul 31, 2020 · dfe03a1 · dfe03a1
1 parent d480508
commit dfe03a1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/A-基础教程/A2-神经网络基本原理简明教程/SourceCode/ch09-NonLinearRegression/HelperClass2/NeuralNet_2_0.py b/A-基础教程/A2-神经网络基本原理简明教程/SourceCode/ch09-NonLinearRegression/HelperClass2/NeuralNet_2_0.py
@@ -147,7 +147,7 @@ def Test(self, dataReader):
         x,y = dataReader.GetTestSet()
         self.forward(x)
         correct = self.__CalAccuracy(self.output, y)
-        print(correct)
+        return correct
 
     def __CalAccuracy(self, a, y):
         assert(a.shape == y.shape)
@@ -181,4 +181,4 @@ def ShowTrainingHistory(self):
         self.loss_trace.ShowLossHistory(self.hp)
 
     def GetTrainingHistory(self):
-        return self.loss_trace
+        return self.loss_trace