罗俊辉 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())