首页 > 解决方案 > OneHotEncoding 分类数据

问题描述

我正在尝试使用 OneHotEncoding 来转换我的 csv 文件的第二列,其中包含公司名称。

from sklearn.preprocessing import OneHotEncoder
ct = ColumnTransformer(transformers=[('encoder', OneHotEncoder(), [1])]), remainder='passthrough'
X = np.array(ct.fit_transform(X))

我收到了这个错误:

    ct = ColumnTransformer(transformers=[('encoder', OneHotEncoder(), [1])]), remainder='passthrough'
                  ^
SyntaxError: can't assign to function call

我哪里做错了?我知道我写 ct 行的方式可能有错误,但我不知道我到底做错了什么。

标签: pythonscikit-learnneural-networkone-hot-encoding

解决方案


我认为你错误地设置了一个括号,它可能应该是这样的:

ct = ColumnTransformer(transformers=[('encoder', OneHotEncoder(), [1])], remainder='passthrough')

推荐阅读