在for循环中使用Sklearn的LabelEncoder错误

负相关性

我正在使用spyder 2运行Python 3,并且在尝试运行此代码时:

from sklearn.preprocessing import LabelEncoder
cv=train.dtypes.loc[train.dtypes=='object'].index
print (cv)

le=LabelEncoder()
for i in cv:
    train[i]=le.fit_transform(train[i])
    test[i]=le.fit_transform(test[i])

我收到此错误:

le=LabelEncoder()
for i in cv:
    train[i]=le.fit_transform(train[i])
    test[i]=le.fit_transform(test[i])


Traceback (most recent call last):

  File "<ipython-input-5-8739984f61b2>", line 3, in <module>
    train[i]=le.fit_transform(train[i])

  File "C:\Users\myname\Anaconda3\lib\site-packages\sklearn\preprocessing\label.py", line 127, in fit_transform
    self.classes_, y = np.unique(y, return_inverse=True)

  File "C:\Users\myname\Anaconda3\lib\site-packages\numpy\lib\arraysetops.py", line 195, in unique
    perm = ar.argsort(kind='mergesort' if return_index else 'quicksort')

TypeError: unorderable types: str() > float()

奇怪的是,如果我在数据中的指定列上调用编码器,则输出成功。例如:

le.fit_transform(test['Race'])

结果是:

le.fit_transform(test['Race'])
Out[7]: array([2, 4, 4, ..., 4, 1, 4], dtype=int64)

我试过了:

float(le.fit_transform(train[i]))
str(le.fit_transform(train[i]))

两者都不起作用。

有人可以帮我吗?

马文·塔森伯格(Marvin Taschenberger)

也有同样的问题。原来我错过了检查缺少的值。检查是否还有剩余(针对您的情况):

print(train.apply(lambda x : sum(x.isnull())))
print(test.apply(lambda x : sum(x.isnull())))

如果您有一些用参数(均值,中值,模数...)替换它们,或者只是将它们编码为字符串,即对于任意变量VAR:

parameter = train[VAR].mean() # parameter = "Nan"
train[VAR].fillna(parameter, inplace = True ) 

本文收集自互联网,转载请注明来源。

如有侵权,请联系 [email protected] 删除。

编辑于
0

我来说两句

0 条评论
登录 后参与评论

相关文章