网络编程
位置:首页>> 网络编程>> Python编程>> python的scikit-learn将特征转成one-hot特征的方法

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

作者:yan456jie  发布时间:2022-06-21 11:34:29 

标签:python,scikit,learn

如下所示:


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()

来源:https://blog.csdn.net/Yan456jie/article/details/72251543

0
投稿

猜你喜欢

手机版 网络编程 asp之家 www.aspxhome.com