pytorch-tabular/pytorch_tabular
A unified framework for Deep Learning Models on tabular data
解決する課題
PyTorch Tabular は、ディープラーニングアーキテクチャを表形式データに適用するための統一されたハイレベル API を提供し、これらの複雑なモデルを実務でも学術研究でもより手軽に利用できるようにします。表形式専用のニューラルネットワークを手動で実装する際の摩擦を取り除きます。
仕組み
PyTorch と PyTorch Lightning 上に構築されたこのライブラリは、トレーニング、評価、予測のための標準化されたインターフェースを提供します。ユーザーはカラム用に DataConfig、特定のアーキテクチャ用に ModelConfig、学習プロセス用に TrainerConfig を定義します。これにより、パイプラインの他の部分を書き換えることなく、表形式データ向けのさまざまなディープラーニングモデル間を簡単に切り替えることができます。
対象者
従来の勾配ブースティングマシンではなく、表形式データセットにディープラーニングを活用したいデータサイエンティストや機械学習研究者、あるいは構造化データ向けのさまざまなニューラルネットワークアーキテクチャを迅速にプロトタイプしベンチマークしたい方々。
ハイライト
- 幅広いモデル選択：TabNet、TabTransformer、FT Transformer、NODE、GANDALF など多様なアーキテクチャを含む。
- スケーラビリティ：PyTorch Lightning を活用し、CPU と GPU 間で自動ロギングしながらトレーニングをスケールさせる。
- カスタマイズ性：専用チュートリアルとフレームワークにより、カスタムモデルの実装をサポート。
- 半教師あり学習：ノイズデータから頑健な特徴表現を学習するための Denoising AutoEncoder を搭載。