TabTransformer 多类分类

Question

TabTransformer：是否有在 Tensorflow 中使用 TabTransformer 进行多类表格数据分类的文档或示例实现？

以下来自 keras 的参考仅适用于二元分类： https://keras.io/examples/structed_data/tabtransformer/

将 sigmoid 转换为 softmax 并将输出层节点增加到 C（总类数）会出现错误。

Answer 1

我修改了您提到的 Keras TabTransformer 代码，并将其改编为多类方法。为了实现这一目标，我使用了 NSL-KDD 入侵检测数据集，该数据集由 5 个主要类别（dos、normal、probe、r2l、u2r）组成，每个类别都包含各种类别。您可以在以下位置找到解决方案实施 TabTransformer 多类分类

我认为这可以很容易地适应不同的多类问题。 我希望这对你有帮助。