diff --git a/combo/config.multitask.template.jsonnet b/combo/config.multitask.template.jsonnet index cf91ef12da1e558fe4ea39679e8c24a497f3c9ae..3553bc11cda8dd4445d07b7a88f4beb1a89a494b 100644 --- a/combo/config.multitask.template.jsonnet +++ b/combo/config.multitask.template.jsonnet @@ -199,7 +199,7 @@ assert pretrained_tokens == null || pretrained_transformer_name == null: "Can't data_loader: { type: "multitask", scheduler: { - batch_size: 10 + batch_size: 20 }, shuffle: true, // batch_sampler: { diff --git a/combo/models/model.py b/combo/models/model.py index 883934eb5e041c8eb94a107ea21bbd89da5b6bbf..dd3629ad803fdb79662a7346340248df85230c1e 100644 --- a/combo/models/model.py +++ b/combo/models/model.py @@ -38,8 +38,7 @@ class NERModel(heads.Head): super().__init__(vocab) self.feedforward_predictor = feedforward_predictor self._accuracy_metric = allen_metrics.CategoricalAccuracy() - self._f1_metric = allen_metrics.SpanBasedF1Measure(vocab, tag_namespace="ner_labels", label_encoding="IOB1", - ignore_classes=["_"]) + self._f1_metric = allen_metrics.SpanBasedF1Measure(vocab, tag_namespace="ner_labels", label_encoding="IOB1") self._loss = 0.0 def forward(self,