DataConversionWarning： Data with input dtype int32, int64 were all converted to float64 by StandardS

╰半夏微凉° 2022-11-24 12:58 221阅读 0赞

机器学习过程中构建一颗决策树对数据进行标准化，使用下面两段代码

  1. X_train = ss_X.fit_transform(X_train)
  2. X_test = ss_X.transform(X_test)

运行之后出现下面警告

  1. D:\Python\Anaconda3\lib\site-packages\sklearn\preprocessing\data.py:625: DataConversionWarning: Data with input dtype int32, int64 were all converted to float64 by StandardScaler.
  2.   return self.partial_fit(X, y)
  3. D:\Python\Anaconda3\lib\site-packages\sklearn\base.py:462: DataConversionWarning: Data with input dtype int32, int64 were all converted to float64 by StandardScaler.
  4.   return self.fit(X, **fit_params).transform(X)

【解决方案】

将代码改成如下所示就可以

  1. X_train = ss_X.fit_transform(X_train.astype(float))
  2. X_test = ss_X.transform(X_test.astype(float))

再次运行就不会出现这样的警告了，希望对你有所帮助。

发表评论

表情：
评论列表 （有 0 条评论，221人围观）

还没有评论，来说两句吧...

相关阅读