PyTorch Tabular：一个面向表格数据的深度学习框架
机器学习
2021-04-29 v1
摘要
尽管深度学习在文本和图像等模态中展现出惊人的有效性，但在表格数据上，其流行度与性能始终落后于梯度提升方法。但近来出现了专门为表格数据设计的新模型，正在推高性能上限。然而流行度仍是一大挑战，因为目前还没有像 Sci-Kit Learn 那样易于使用、开箱即用的深度学习库。PyTorch Tabular 是一个新的深度学习库，它让深度学习和表格数据的处理变得轻松而快速。该库构建于 PyTorch 和 PyTorch Lightning 之上，可直接处理 pandas 数据框。许多 SOTA 模型（如 NODE 和 TabNet）已通过统一 API 集成并实现于该库中。PyTorch Tabular 的设计兼顾了研究人员的易扩展性、从业者的易用性，以及工业部署的鲁棒性。
引用
@article{arxiv.2104.13638,
title = {PyTorch Tabular: A Framework for Deep Learning with Tabular Data},
author = {Manu Joseph},
journal= {arXiv preprint arXiv:2104.13638},
year = {2021}
}
备注
This work has been submitted to the IEEE for possible publication