You are right. It’s a binary classification task, so the label is saved as type long. At first, I just tried to fix the problem during taining stage. But it didn’t work, so in the end I changed the dtype of my dataset.