1
我想使用sklearn热门编码器进行状态的热编码。这里是我的熊猫数据框:一个使用sklearn的国家的热门编码
State
0 FL
1 CA
2 MD
3 NY
4 NY
5 NY
6 NY
我写道:
from sklearn.preprocessing import OneHotEncoder
enc=OneHotEncoder(sparse=False)
enc.fit(data)
而这里的错误:
---------------------------------------------------------------------------
ValueError Traceback (most recent call last)
<ipython-input-78-a0b336acd757> in <module>()
----> 1 enc.fit(data)
/anaconda/envs/env3_insight/lib/python3.6/site-packages/sklearn/preprocessing/data.py in fit(self, X, y)
1842 self
1843 """
-> 1844 self.fit_transform(X)
1845 return self
1846
/anaconda/envs/env3_insight/lib/python3.6/site-packages/sklearn/preprocessing/data.py in fit_transform(self, X, y)
1900 """
1901 return _transform_selected(X, self._fit_transform,
-> 1902 self.categorical_features, copy=True)
1903
1904 def _transform(self, X):
/anaconda/envs/env3_insight/lib/python3.6/site-packages/sklearn/preprocessing/data.py in _transform_selected(X, transform, selected, copy)
1695 X : array or sparse matrix, shape=(n_samples, n_features_new)
1696 """
-> 1697 X = check_array(X, accept_sparse='csc', copy=copy, dtype=FLOAT_DTYPES)
1698
1699 if isinstance(selected, six.string_types) and selected == "all":
/anaconda/envs/env3_insight/lib/python3.6/site-packages/sklearn/utils/validation.py in check_array(array, accept_sparse, dtype, order, copy, force_all_finite, ensure_2d, allow_nd, ensure_min_samples, ensure_min_features, warn_on_dtype, estimator)
380 force_all_finite)
381 else:
--> 382 array = np.array(array, dtype=dtype, order=order, copy=copy)
383
384 if ensure_2d:
ValueError: could not convert string to float: 'NY'
我不明白。我认为做一个热门编码的重点就是从分类,通常是字符串信息转换为数字...为什么它说我不能将字符串转换为浮点数,然后呢?
首先,您需要使用LabelEncoder。 – ayhan
OneHotEncoder,[文档建议](http://scikit-learn.org/stable/modules/generated/sklearn.preprocessing.OneHotEncoder.html)只接受数字数据。首先为您的数据分配一些数值,如'FL'为1,'CA'为2等,然后使用OneHotEncoding。按照@ayhan的建议将字符串转换为数字使用LabelEncoder。 –