当前位置:首页 > Python >

python的scikit-learn将特征转成one-hot特征_python sklearn one-hot

发布时间:2017-06-16 08:56:30 作者:佚名 阅读:(815)

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 sklearn one-hot

© 2015-2019 - 吾爱编程网 版权所有 苏ICP备18033726号-1关于我们 - 网站声明 - 联系我们