数据预处理中使用toarray()与onehotencoding

我是机器学习的新手。我有一个疑问:为什么在这里使用onehotencoding时要用toarray(),而使用label encoding时却不需要?我完全不明白,请有人帮帮我。

from sklearn.preprocessing import LabelEncoder, OneHotEncoderlabel_encoder_x = LabelEncoder()x[:, 0] = label_encoder_x.fit_transform(x[:, 0])onehotencoder = OneHotEncoder(categorical_features= [0])x = onehotencoder.fit_transform(x).toarray()label_encoder_y = LabelEncoder()y = label_encoder_y.fit_transform(y)

回答:

正如文档中所述,OneHotEncoder默认会返回一个稀疏矩阵,这种格式不太直观。举个简单的例子:

>>> df   00  21  22  23  14  15  26  27  18  19  1onehotencoder = OneHotEncoder(categorical_features= [0])>>> onehotencoder.fit_transform(df[0].values.reshape(1,-1))<1x10 sparse matrix of type '<class 'numpy.float64'>'    with 10 stored elements in COOrdinate format>

然而,如果你调用toarray,你会得到一个更容易处理或至少更容易理解的结果:

>>> onehotencoder.fit_transform(df[0].values.reshape(1,-1)).toarray()array([[1., 2., 2., 1., 1., 2., 2., 1., 1., 1.]])

另一种方法是在创建OneHotEncoder时使用sparse = False,这样它默认会创建一个密集数组:

onehotencoder = OneHotEncoder(categorical_features= [0], sparse=False)>>> onehotencoder.fit_transform(df[0].values.reshape(1,-1))array([[1., 2., 2., 1., 1., 2., 2., 1., 1., 1.]])

Related Posts

L1-L2正则化的不同系数

我想对网络的权重同时应用L1和L2正则化。然而,我找不…

使用scikit-learn的无监督方法将列表分类成不同组别,有没有办法?

我有一系列实例,每个实例都有一份列表,代表它所遵循的不…

f1_score metric in lightgbm

我想使用自定义指标f1_score来训练一个lgb模型…

通过相关系数矩阵进行特征选择

我在测试不同的算法时,如逻辑回归、高斯朴素贝叶斯、随机…

可以将机器学习库用于流式输入和输出吗?

已关闭。此问题需要更加聚焦。目前不接受回答。 想要改进…

在TensorFlow中,queue.dequeue_up_to()方法的用途是什么?

我对这个方法感到非常困惑,特别是当我发现这个令人费解的…

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注