andrea.pedrotti
/
gFun
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Implemented custom micro and macro F1 in pl (cpu and gpu) + various TODO

This commit is contained in:
andrea 2021-01-20 14:57:31 +01:00
parent faa387f696
commit d6eeabe6ab
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
refactor/models/pl_gru.py
View File

@ -11,6 +11,7 @@ from models.helpers import init_embeddings
from util.pl_metrics import CustomF1 from util.pl_metrics import CustomF1
from util.evaluation import evaluate from util.evaluation import evaluate
# TODO: it should also be possible to compute metrics independently for each language!
class RecurrentModel(pl.LightningModule): class RecurrentModel(pl.LightningModule):
""" """