wassname/Castor
1
0
Fork 0
mirror of https://github.com/wassname/Castor.git synced 2026-09-09 11:13:20 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update Kim CNN SST-1 dataset to use torchtext 0.2.0 (#81)

Browse Source
This commit is contained in:
Michael Tu
2017-11-07 21:37:18 -05:00
committed by GitHub
parent 2344354cbf
commit aaff291edb
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
kim_cnn/SST1.py
+2 -2
View File
@@ -10,6 +10,6 @@ class SST1Dataset(data.TabularDataset):
prefix_name = 'stsa.fine.'
path = './data'
return super(SST1Dataset, cls).splits(
os.path.join(path, prefix_name), train, validation, test,
path, train=prefix_name + train, validation=prefix_name + validation, test=prefix_name + test,
format='TSV', fields=[('label', label_field), ('text', text_field)]
)
)
kim_cnn/main.py
+1 -1
View File
@@ -71,4 +71,4 @@ def predict(dataset_iter, dataset, dataset_name):
predict(dataset_iter=dev_iter, dataset=dev, dataset_name="valid")
# Run the model on the test set
predict(dataset_iter=test_iter, dataset=test, dataset_name="test")
predict(dataset_iter=test_iter, dataset=test, dataset_name="test")