Upload Plt.py

Plt.py

@@ -0,0 +1,35 @@
+from matplotlib import pyplot as plt
+
+
+def Plt(file):
+    filereader = open(file.name, 'r')
+    # 可视化
+    Loss_list = []
+    Accuracy_list = []
+
+    for line in filereader.readlines():
+        if line[0:4] == "loss":
+            list = line.split()
+            # print(list[1])
+            Loss_list.append(float(list[1]))
+            Accuracy_list.append(float(list[3]))
+
+    print(Loss_list)
+    length = len(Loss_list)
+    x1 = range(0, length)
+    x2 = range(0, length)
+    y1 = Accuracy_list
+    # y2 = Loss_list[4:]
+    y2 = Loss_list
+    plt.subplot(2, 1, 1)
+    plt.plot(x1, y1)
+    plt.title('Test accuracy vs. epoches')
+    plt.ylabel('Test accuracy')
+    plt.subplot(2, 1, 2)
+    plt.plot(x2, y2)
+    plt.xlabel('Test loss vs. epoches')
+    plt.ylabel('Test loss')
+    plt.savefig("result.jpg")
+    # plt.show()
+    return "result.jpg"
+