Python CrossEntropyLoss Examples

Example #1

File: customized_data.py Project: rpatil524/cogdl

def _get_loss_fn(metric):
    if metric in ("accuracy", "multiclass_f1"):
        return CrossEntropyLoss()
    elif metric == "multilabel_f1":
        return BCEWithLogitsLoss()
    else:
        raise NotImplementedError

Example #2

File: dataset.py Project: rpatil524/cogdl

 def get_loss_fn(self):
     return CrossEntropyLoss()