python - ValueError：无法将字符串转换为浮点数：'France'

Question

在 Python 中使用 oneHotEncoder 时出错。

我在 Udemy 课程之后使用 Python 学习机器学习。当我尝试使用 oneHotEncorder 对源数据中的国家名称进行编码时，出现以下错误。

array = np.array(array, dtype=dtype, order=order, copy=copy)
ValueError: could not convert string to float: 'France'

代码：

from sklearn.preprocessing import OneHotEncoder
oneHotEncObj=OneHotEncoder(categorical_features=[0])
X=oneHotEncObj.fit_transform(X).toarray()

标签： pythonmachine-learningscikit-learn

Answer 1

您正在寻找：https ://scikit-learn.org/stable/modules/generated/sklearn.preprocessing.LabelEncoder.html

尝试将 onehotencoder 替换为preprocessing.LabelEncoder()