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를 포함합니다.