Update trainer.py
Browse files- 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 |
-
|
| 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=
|
| 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,
|