我要评论问题
最近用深度学习训练一个分类模型，在划分训练集和测试集时遇到一个问题。
意思是list的索引必须是整形，不能是float型。
经排查发现
存储标签的list是使用np.empty()函数创建的
这个函数可以创建指定shape的list，但是创建的并非是一个空的多维数组
如 np.empty((3,3))，默认为numpy.float64型，而实际标签值为整形
实际返回结果为：
print(np.empty((3,3))) [[6.95177882e-310 6.95177882e-310 0.00000000e+000] [0.00000000e+000 6.95177883e-310 0.00000000e+000] [0.00000000e+000 0.00000000e+000 1.21698002e-152]]
查阅资料，发现np.empty()返回一个随机元素的矩阵，大小按照参数定义。
解决方法
其实使用在使用np.empty()创建list，可以指定类型
如：
print(np.empty(3,3,dtype=int)) [[4128860 6029375 3801156] [4391004 6553711 7536741] [5242972 7602297 7274600]]
总结
这样就搞定了！
以上为个人经验，希望能给大家一个参考，也希望大家多多支持代码网。
发表评论