@Sourena_Yadegari
Isn’t it also necessary to reindex the data to do something useful with this dataset?
For example the CrossEntropy criterion expects a class index in the range [0, C-1].
So, in this case 1,2 have to be reindexed as 0,1 otherwise the criterion will throw an error.