您的位置:首页 > 编程语言 > Python开发

python的scikit-learn将特征转成one-hot特征的方法

2018-07-10 10:51 537 查看

如下所示:

enc = OneHotEncoder(categorical_features=np.array([0,1,2]),n_values=[5,4,2])
enc.fit(train_data)
train_data = enc.transform(train_data).toarray()
test_data = enc.transform(test_data).toarray()

以上这篇python的scikit-learn将特征转成one-hot特征的方法就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持脚本之家。

您可能感兴趣的文章:

内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  python scikit learn