【机器学习算法应用与学习_3_代码API篇】2.0预处理

【机器学习算法应用与学习_3_代码API篇】2.0预处理

编码

独热编码

基于树的算法不需要独热编码。

1.pandas方法

from sklearn.datasets import load_iris
import pandas as pd
#创建数据集
data = pd.DataFrame({'one':[1,2,3],'two':[2,3,4],'city':[3,4,5]})
#对city独热编码
data =pd.get_dummies(data, columns=['city'])
data

View Code

 2.sklearn方法

from sklearn import preprocessing
enc = preprocessing.OneHotEncoder()
enc.fit([[0, 0, 3], [1, 1, 0], [0, 2, 1], [1, 0, 2]])    # fit来学习编码
enc.transform([[0, 1, 3]]).toarray()    # 进行编码

View Code

 

https://zhuanlan.zhihu.com/p/35287916

posted on 2019-08-12 19:18  everda 阅读( ...) 评论( ...) 编辑 收藏

转载于:https://www.cnblogs.com/everda/p/11342020.html