コード例 #1
0
ファイル: customized_data.py プロジェクト: rpatil524/cogdl 
def _get_loss_fn(metric):
    if metric in ("accuracy", "multiclass_f1"):
        return CrossEntropyLoss()
    elif metric == "multilabel_f1":
        return BCEWithLogitsLoss()
    else:
        raise NotImplementedError
コード例 #2
0
ファイル: dataset.py プロジェクト: rpatil524/cogdl 
 def get_loss_fn(self):
     return CrossEntropyLoss()