From d6eeabe6abd17545031fe4c1f341dc0e85cb5349 Mon Sep 17 00:00:00 2001 From: andrea Date: Wed, 20 Jan 2021 14:57:31 +0100 Subject: [PATCH] Implemented custom micro and macro F1 in pl (cpu and gpu) + various TODO --- refactor/models/pl_gru.py | 1 + 1 file changed, 1 insertion(+) diff --git a/refactor/models/pl_gru.py b/refactor/models/pl_gru.py index c810220..9883b92 100644 --- a/refactor/models/pl_gru.py +++ b/refactor/models/pl_gru.py @@ -11,6 +11,7 @@ from models.helpers import init_embeddings from util.pl_metrics import CustomF1 from util.evaluation import evaluate +# TODO: it should also be possible to compute metrics independently for each language! class RecurrentModel(pl.LightningModule): """