Ejemplo n.º 1
0
def _get_loss_fn(metric):
    if metric in ("accuracy", "multiclass_f1"):
        return CrossEntropyLoss()
    elif metric == "multilabel_f1":
        return BCEWithLogitsLoss()
    else:
        raise NotImplementedError
Ejemplo n.º 2
0
 def get_loss_fn(self):
     return CrossEntropyLoss()