yeomtong commited on
Commit
4a6fe36
·
verified ·
1 Parent(s): f1c3b7e

Update trainer.py

Browse files
Files changed (1) hide show
  1. trainer.py +2 -2
trainer.py CHANGED
@@ -186,7 +186,7 @@ if __name__ == "__main__":
186
 
187
  # read values from cfg as usual:
188
  conll_train_path = cfg["data"]["conll_train"]
189
- conll_dev_path = cfg["data"].get("conll_dev")
190
  conll_test_path = cfg["data"].get("conll_test")
191
  word_col_idx = cfg["data"]["word_col_idx"]
192
  srl_first_col_idx= cfg["data"]["srl_first_col_idx"]
@@ -223,7 +223,7 @@ if __name__ == "__main__":
223
  train_bf_loader, dev_bf_loader, label2id, id2label = \
224
  data_processing_for_loader_conll(
225
  train_conll=conll_train_path,
226
- dev_conll=conll_dev_path,
227
  # test_conll=conll_test_path,
228
  tokenizer=tokenizer,
229
  word_col_idx=word_col_idx,
 
186
 
187
  # read values from cfg as usual:
188
  conll_train_path = cfg["data"]["conll_train"]
189
+ conll_valid_path = cfg["data"].get("conll_valid")
190
  conll_test_path = cfg["data"].get("conll_test")
191
  word_col_idx = cfg["data"]["word_col_idx"]
192
  srl_first_col_idx= cfg["data"]["srl_first_col_idx"]
 
223
  train_bf_loader, dev_bf_loader, label2id, id2label = \
224
  data_processing_for_loader_conll(
225
  train_conll=conll_train_path,
226
+ dev_conll=conll_vald_path,
227
  # test_conll=conll_test_path,
228
  tokenizer=tokenizer,
229
  word_col_idx=word_col_idx,