torchtext不支持對數據進行拆分

train_path = '../data/match_data/train.csv'
test_path = '../data/match_data/dev.csv'
all_train_data = pd.read_csv(train_path)
train, val = train_test_split(all_train_data, test_size=0.2)
train.to_csv("../data/match_data/split_train.csv", index=False)
val.to_csv("../data/match_data/split_val.csv", index=False)

數據拆分代碼見上面代碼

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章