首页 > 编程 > Python > 正文

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

2020-02-15 22:18:56
字体:
来源:转载
供稿:网友

如下所示:

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特征的方法就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持武林站长站。

发表评论 共有条评论
用户名: 密码:
验证码: 匿名发表