Wrong Test file path fixed

isaac
Sai Pavan Mandava 5 years ago
parent 3916df452f
commit 3455af4e40

@ -48,10 +48,10 @@
},
"trainer": {
"optimizer": {
"type": "adam",
"type": "adagrad",
"lr": 0.001
},
"num_epochs": 2,
"cuda_device": 1
"num_epochs": 20,
"cuda_device": 3
}
}

@ -51,8 +51,8 @@ def make_predictions(model: Model, dataset_reader: DatasetReader, file_path: str
def load_model_and_run_predictions(saved_model_dir: str):
project_root = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
train_file_path = project_root + '/data/tsv/train.tsv'
test_file_path = project_root + '/data/tsv/test.tsv'
dev_file_path = project_root + '/data/jsonl/dev.jsonl'
test_file_path = project_root + '/data/jsonl/test.jsonl'
model_archive = load_archive(os.path.join(saved_model_dir, 'model.tar.gz'))
citation_dataset_reader = CitationDataSetReader()

Loading…
Cancel
Save