罗俊辉 1 anno fa
parent
commit
6312409b1a
1 ha cambiato i file con 2 aggiunte e 2 eliminazioni
  1. 2 2
      main.py

+ 2 - 2
main.py

@@ -10,8 +10,8 @@ from sklearn.metrics import accuracy_score
 
 with open("whole_data/x_data.json") as f1:
     x_list = json.loads(f1.read())
-    X_train = np.array(x_list[:10000])
-    X_test = np.array(x_list[10000:])
+    X_train = x_list[:10000]
+    X_test = x_list[10000:]
 
 with open("whole_data/y_data.json") as f2:
     y_list = json.loads(f2.read())