solution: use. Astype (‘ int ‘) to convert a label to an int

Reference:

https://blog.csdn.net/qq_41870817/article/details/89419726