From aaff291edba2e69fb06a72a214bdf7d641b9d9d0 Mon Sep 17 00:00:00 2001 From: Michael Tu Date: Tue, 7 Nov 2017 21:37:18 -0500 Subject: [PATCH] Update Kim CNN SST-1 dataset to use torchtext 0.2.0 (#81) --- kim_cnn/SST1.py | 4 ++-- kim_cnn/main.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/kim_cnn/SST1.py b/kim_cnn/SST1.py index e8da083..30b92c1 100644 --- a/kim_cnn/SST1.py +++ b/kim_cnn/SST1.py @@ -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)] - ) \ No newline at end of file + ) diff --git a/kim_cnn/main.py b/kim_cnn/main.py index c0f695f..cd9a7b6 100644 --- a/kim_cnn/main.py +++ b/kim_cnn/main.py @@ -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") \ No newline at end of file +predict(dataset_iter=test_iter, dataset=test, dataset_name="test")