How to use class torch.nn.MultiLabelMarginLoss?

1

Hi everyone,
I’m trying to use torch.nn.MultiLabelMarginLoss as a 1999 multiclass text classification proble. However, the docs really confuses me.
The docs is below:
Creates a criterion that optimizes a multi-class multi-classification hinge loss (margin-based loss) between input x (a 2D mini-batch Tensor) and output y (which is a 2D Tensor of target class indices). For each sample in the mini-batch:

loss(x, y) = sum_ij(max(0, 1 - (x[y[j]] - x[i]))) / x.size(0)
where i == 0 to x.size(0), j == 0 to y.size(0), y[j] != 0, and i != y[j] for all i and j.

y and x must have the same size.

The criterion only considers the first non zero y[j] targets.

This allows for different samples to have variable amounts of target classes

Could anyone tell me what’s the appropriate y in the argument when a sample is tagged as classes (1,100,200)?

2 Likes
2

did you ever found a solution to this?

3

I believe relative code can be found here:

Very confusing indeed, Don’t see why y and x must have the same size, might be for parallellism.
I’m looking into, might come back to you if I find something meaningful.