From 85e5603d7b5b7a9fa54f5307f6e993d063c0718e Mon Sep 17 00:00:00 2001
From: plor4hs9e <2432543846@qq.com>
Date: Sun, 28 Apr 2024 20:42:02 +0800
Subject: [PATCH] ADD file via upload

---
 vision.py | 42 ++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 42 insertions(+)
 create mode 100644 vision.py

diff --git a/vision.py b/vision.py
new file mode 100644
index 0000000..804af8b
--- /dev/null
+++ b/vision.py
@@ -0,0 +1,42 @@
+import matplotlib.pyplot as plt
+
+accuracy_0_list = []
+accuracy_1_list = []
+
+with open('vision/text-loss-0', 'r') as fp0:
+    accuracy0 = fp0.readline().replace('\n', '')
+    while accuracy0:
+        accuracy_0_list.append(float(accuracy0))
+        accuracy0 = fp0.readline().replace('\n', '')
+
+with open('vision/text-loss-1', 'r') as fp1:
+    accuracy1 = fp1.readline().replace('\n', '')
+    while accuracy1:
+        accuracy_1_list.append(float(accuracy1))
+        accuracy1 = fp1.readline().replace('\n', '')
+
+
+step_0_list = [i for i in range(len(accuracy_0_list))]
+step_1_list = [j for j in range(len(accuracy_1_list))]
+
+plt.subplot(2,2,1)
+plt.plot(step_0_list, accuracy_0_list, 'o-',label='loss-0', color='red')
+plt.plot(step_1_list, accuracy_1_list, 'o-', label='loss', color='green')
+plt.xlabel("step")
+plt.ylabel("test-loss")
+plt.legend(loc='lower right')
+
+plt.subplot(2,2,2)
+plt.plot(step_0_list, accuracy_0_list,'o-', label='loss-0', color='red')
+plt.xlabel("step")
+plt.ylabel("test-loss")
+plt.legend(loc='lower right')
+
+plt.subplot(2,2,3)
+plt.plot(step_1_list, accuracy_1_list, 'o-', label='loss-1', color='green')
+plt.xlabel("step")
+plt.ylabel("test-loss")
+plt.legend(loc='lower right')
+
+plt.show()
+