issue: 319

JdYwz · Sep 4, 2022 · fa46676 · fa46676
1 parent 8e680d3
commit fa46676
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pycorrector/macbert/preprocess.py b/pycorrector/macbert/preprocess.py
@@ -193,7 +193,7 @@ def main():
     print('data_size:', len(rst_items))
     train_lst, dev_lst = train_test_split(rst_items, test_size=0.1, random_state=42)
     save_json(train_lst, os.path.join(pwd_path, 'output/train.json'))
-    save_json(train_lst, os.path.join(pwd_path, 'output/dev.json'))
+    save_json(dev_lst, os.path.join(pwd_path, 'output/dev.json'))
     save_json(test_lst, os.path.join(pwd_path, 'output/test.json'))