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 @@
},
"model": {
"type": "simple_tagger",
"calculate_span_f1": true,
"label_encoding": "BIOUL",
"encoder": {
"type": "lstm",
"bidirectional": true,
......@@ -67,13 +69,14 @@
"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",
"trainer": {
"cuda_device": 1,
"cuda_device": -1,
"grad_norm": 5,
"num_epochs": 4,
"num_serialized_models_to_keep": 3,
"num_serialized_models_to_keep": 1,
"optimizer": {
"type": "bert_adam",
"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