Implementation of Squared Earth movers distance loss function for ordinal scale

vision
2

You could try to translate this code by using the torch namespace.
E.g this might work (untested):

def earth_mover_distance(y_true, y_pred):
    return torch.mean(torch.square(torch.cumsum(y_true, dim=-1) - torch.cumsum(y_pred, dim=-1)), dim=-1)
1 Like