I am dividing a set of image data into five categories. I want to set the label like this, because these data are related. Can you tell me how to modify the related code?My original label is [0, 1, 2, 3, 4]
1 1 0 0 0 0
2 1 1 0 0 0
3 1 1 1 0 0
4 1 1 1 1 0
5 1 1 1 1 1
Thanks！