关于pycharm的Input contains NaN, infinity or a value too large for dtype(‘float64‘)问题。

关于pycharm的Input contains NaN, infinity or a value too large for dtype(‘float64’)问题。

最近在做机器学习模型训练时，提示Input contains NaN, infinity or a value too large for dtype(‘float64’)错误。如下图所示：

起初以为是训练集中含有空值，就去检查了训练集中的数据，但是并没有解决这个问题。后来debug发现，标签中的数字都变为了float64位类型，于是想到将标签中的数字转为整数。

y_train = numpy.uint8(y_train)
y_test = numpy.uint8(y_test)

再次进行训练成功运行

目前网上关于这个问题大都由训练集中是否含有空缺值，是否含有无穷值两个方面出发，几乎没有提及是数据格式问题，仅在此记录一下。
第一次发博客，如有问题还请各位大佬指正。