浏览代码

获取 1 , 2月 label

罗俊辉 1 年之前
父节点
当前提交
8f1f6d7df3
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 1 1
      main.py

+ 1 - 1
main.py

@@ -59,7 +59,7 @@ with open("whole_data/x_data.json") as f1:
 
 
 with open("whole_data/y_data.json") as f2:
 with open("whole_data/y_data.json") as f2:
     y_list = json.loads(f2.read())
     y_list = json.loads(f2.read())
-    y__list = [0 if i <= 25 else 1 for i in y_list]
+    y__list = [0 if i <= 56 else 1 for i in y_list]
     y_train = np.array(y__list[:86434])
     y_train = np.array(y__list[:86434])
     y_test = np.array(y__list[86434:])
     y_test = np.array(y__list[86434:])