TabTransformer: Tabular Data Modeling Using Contextual Embeddings

[upura]

どんなもの？

カテゴリ変数の処理にTransformer層を用いた「TabTransformer」の提案。論文内の実験では、GBDTに匹敵する性能を示している。

論文リンク

https://arxiv.org/abs/2012.06678