我尝试在sklearn中进行插补,但出现错误

时间:2020-08-11 18:53:49

标签: scikit-learn imputets

我尝试下面的代码,但是有一些错误。

imp=SimpleImputer(missing_values='NaN',strategy="mean")
col = veriler.iloc[:,1:4].values
type(col) ##numpy.ndarray
imp=imp.fit(col)

ValueError:输入包含NaN,无穷大或对于dtype('float64')而言太大的值。

1 个答案:

答案 0 :(得分:0)

您需要将无穷大值转换为有界值才能应用插补。 np.nan_to_numnaninf-inf剪辑为可行的值。

例如:

import numpy as np
from sklearn.impute import SimpleImputer
imp_mean = SimpleImputer(missing_values=np.nan, strategy='mean')
X = [[7, np.inf, 3], [4, np.nan, 6], [10, 5, 9]]
X = np.nan_to_num(X, nan=-9999, posinf=33333333, neginf=-33333333)
imp_mean.fit(X)
>>> SimpleImputer(add_indicator=False, copy=True, fill_value=None,
              missing_values=nan, strategy='mean', verbose=0)

对于变换,也可以应用:

X = [[np.nan, 2, 3], [4, np.nan, 6], [10, np.nan, 9], [np.nan, np.inf, -np.inf]]
X = np.nan_to_num(X, nan=-9999, posinf=33333333, neginf=-33333333)
print(imp_mean.transform(X))

>>>
[[-9.9990000e+03  2.0000000e+00  3.0000000e+00]
 [ 4.0000000e+00 -9.9990000e+03  6.0000000e+00]
 [ 1.0000000e+01 -9.9990000e+03  9.0000000e+00]
 [-9.9990000e+03  3.3333333e+07 -3.3333333e+07]]