罗俊辉 1 年之前
父節點
當前提交
6312409b1a
共有 1 個文件被更改,包括 2 次插入2 次删除
  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())