栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

ValueError: X.dtype should be np.float32, got float64

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

ValueError: X.dtype should be np.float32, got float64

背景

在用GBDT系列训练时,报错ValueError: X.dtype should be np.float32, got float64,如下所示。

ValueError                                Traceback (most recent call last)
 in ()
----> 1 abc.apply(X_train)

~/tmp/dataset/Augboost+FM/AugBoost.py in apply(self, X)
    461             for j in range(n_classes):
    462                 estimator = self.estimators_[i, j]
--> 463                 leaves[:, i, j] = estimator.apply(np.concatenate([X_original, X], axis=1), check_input=False)
    464 
    465         return leaves

~/anaconda3/lib/python3.7/site-packages/sklearn/tree/tree.py in apply(self, X, check_input)
    464         check_is_fitted(self, 'tree_')
    465         X = self._validate_X_predict(X, check_input)
--> 466         return self.tree_.apply(X)
    467 
    468     def decision_path(self, X, check_input=True):

sklearn/tree/_tree.pyx in sklearn.tree._tree.Tree.apply()

sklearn/tree/_tree.pyx in sklearn.tree._tree.Tree.apply()

sklearn/tree/_tree.pyx in sklearn.tree._tree.Tree._apply_dense()

ValueError: X.dtype should be np.float32, got float64

解决方法

很显然,就是字面上的意思,只能是np.float32,但是给出的是float64
我看了tree.py

sklearn内置的代码一路走下来就应该是32位的。所以怀疑是自己前面int类型的输入训练集在转化是转化为了64位

看了一下,前面有这样的代码

X_original = X
X_normed = self.normalizer.transform(X)

X是我输入的X_train ,dataframe格式,int类型的数据

看下X_normed

ok,发现问题了,经过normalizer.transform()我的数据变成了numpy.ndarray类型,float64。那么咱们把类型转化过来就行了

ndarray的数据类型:

https://blog.csdn.net/weixin_43181110/article/details/83996915?spm=1001.2101.3001.6650.1&utm_medium=distribute.pc_relevant.none-task-blog-2%7Edefault%7ECTRLIST%7Edefault-1.no_search_link&depth_1-utm_source=distribute.pc_relevant.none-task-blog-2%7Edefault%7ECTRLIST%7Edefault-1.no_search_link

X_normed = X_normed.astype(np.float32)

这样就可以啦

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/503966.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号