脚本之家,脚本语言编程技术及教程分享平台!
分类导航

Python|VBS|Ruby|Lua|perl|VBA|Golang|PowerShell|Erlang|autoit|Dos|bat|

服务器之家 - 脚本之家 - Python - python的scikit-learn将特征转成one-hot特征的方法

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

2021-03-15 00:23yan456jie Python

今天小编就为大家分享一篇python的scikit-learn将特征转成one-hot特征的方法,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧

如下所示:

?
1
2
3
4
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特征的方法就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持服务器之家。

原文链接:https://blog.csdn.net/Yan456jie/article/details/72251543

延伸 · 阅读

精彩推荐