如下所示:
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