Skip to content
Snippets Groups Projects
Commit c20a29ea authored by muller's avatar muller :art:
Browse files

added F1 measure for evaluation during training

parent 068f5571
Branches
No related tags found
No related merge requests found
...@@ -22,6 +22,8 @@ ...@@ -22,6 +22,8 @@
}, },
"model": { "model": {
"type": "simple_tagger", "type": "simple_tagger",
"calculate_span_f1": true,
"label_encoding": "BIOUL",
"encoder": { "encoder": {
"type": "lstm", "type": "lstm",
"bidirectional": true, "bidirectional": true,
...@@ -67,13 +69,14 @@ ...@@ -67,13 +69,14 @@
"train_data_path": "../data/eng.sdrt.stac/eng.sdrt.stac_train.ner.conllu", "train_data_path": "../data/eng.sdrt.stac/eng.sdrt.stac_train.ner.conllu",
"validation_data_path": "../data/eng.sdrt.stac/eng.sdrt.stac_dev.ner.conllu", "validation_data_path": "../data/eng.sdrt.stac/eng.sdrt.stac_dev.ner.conllu",
"trainer": { "trainer": {
"cuda_device": 1, "cuda_device": -1,
"grad_norm": 5, "grad_norm": 5,
"num_epochs": 4, "num_epochs": 4,
"num_serialized_models_to_keep": 3, "num_serialized_models_to_keep": 1,
"optimizer": { "optimizer": {
"type": "bert_adam", "type": "bert_adam",
"lr": 0.001 "lr": 0.001
} },
"validation_metric": "+f1-measure-overall"
} }
} }
\ No newline at end of file
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment