Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

PyTorch Tabular: A Framework for Deep Learning with Tabular Data

About

In spite of showing unreasonable effectiveness in modalities like Text and Image, Deep Learning has always lagged Gradient Boosting in tabular data - both in popularity and performance. But recently there have been newer models created specifically for tabular data, which is pushing the performance bar. But popularity is still a challenge because there is no easy, ready-to-use library like Sci-Kit Learn for deep learning. PyTorch Tabular is a new deep learning library which makes working with Deep Learning and tabular data easy and fast. It is a library built on top of PyTorch and PyTorch Lightning and works on pandas dataframes directly. Many SOTA models like NODE and TabNet are already integrated and implemented in the library with a unified API. PyTorch Tabular is designed to be easily extensible for researchers, simple for practitioners, and robust in industrial deployments.

Manu Joseph• 2021

Related benchmarks

TaskDatasetResultRank
ClassificationLung
ACC72.95
96
ClassificationAdult
Accuracy83.4
86
ClassificationGLI_85
Accuracy69.14
78
ClassificationColon
Accuracy60.5
78
ClassificationTOX_171
Accuracy57.8
78
ClassificationSMK_CAN_187
Accuracy59.16
72
ClassificationALLAML
Accuracy78.12
72
ClassificationHDLSS Datasets Summary
Average Rank19.67
66
ClassificationProstate_GE
Accuracy45.01
64
ClassificationARCENE
Accuracy54.8
60
Showing 10 of 26 rows

Other info

Follow for update