python

关注公众号 jb51net

关闭
首页 > 脚本专栏 > python > 记录numpy.empty()函数引发的问题

Python记录numpy.empty()函数引发的问题及解决

作者:pangpd

这篇文章主要介绍了Python记录numpy.empty()函数引发的问题及解决方案,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教

问题

最近用深度学习训练一个分类模型,在划分训练集和测试集时遇到一个问题。

意思是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]]

总结

这样就搞定了!

以上为个人经验,希望能给大家一个参考,也希望大家多多支持脚本之家。

您可能感兴趣的文章:
阅读全文