Update data_cut_off.py

wangjiwu · Nov 29, 2020 · fab4aba · fab4aba
1 parent 3d54e21
commit fab4aba
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/data_cut_off.py b/data_cut_off.py
@@ -18,4 +18,7 @@
 
 
     dev_set = pd_all.iloc[0:pd_all.shape[0]/10]
-    dev_set.to_csv("glue/dev.tsv", index=False, sep='\t')
+    train_set = pd_all.iloc[pd_all.shape[0]/10, -1]
+
+    dev_set.to_csv("glue/dev.tsv", index=False, sep='\t')
+    train_set.to_csv("glue/train.tsv", index=False, sep='\t')