Now, I have some data-set(below)
data-set example : Input -> [ [1,6,5,4,7,7], [2,7,2,6,4,4], [6,1,6,2,3,3] ]
Output -> [ [1,0,0], [0,1,0], [0,0,1] ]
Now, I am going to use MultiLabelMarginLoss.
But I don’t know how to use this loss-function.
If this loss function is not suitable for this case, please tell the other loss functions.
What’s the most difficult point for me now is how to use the above loss function(MultiLabelMarginLoss)