一个不同类别的热编码Python

2024-05-29 03:11:26 发布

您现在位置:Python中文网/ 问答频道 /正文

我正在使用人工神经网络进行ML应用。数据列是分类的。这就是为什么我对所有特性使用一种热编码方法。让我问一下我的问题。想象一下我的专栏

ANTENNA_TYPE

1234

3345

3412

5435

5623

我有5个天线类型。当我进行一次热编码时,输出会是

1 0 0 0 0

0 1 0 0 0

0 0 1 0 0

0 0 0 1 0

0 0 0 0 1

之后,下一次训练和预测

当我想预测不同的天线类型为9999时,我不知道如何转换为一个热编码? 有没有人向我推荐这方面的最佳做法

提前谢谢

hot_encoder = OneHotEncoder(categorical_features=[1])
dataset = hot_encoder.fit_transform(dataset).toarray()

Tags: 数据类型编码encoder分类特性mldataset

热门问题