pytorch_tabular
v1.1.1


Pytorch Tabular는 Tabular Data를 사용한 딥 러닝을 실제 사례 및 연구에 모두 쉽고 액세스 할 수 있도록하는 것을 목표로합니다. 도서관 설계의 핵심 원칙은 다음과 같습니다.
그것은 Pytorch (분명히) 및 Pytorch Lightning과 같은 자이언트의 어깨에 세워졌습니다.
설치에는 Pytorch가 포함되어 있지만 가장 좋은 방법은 여기에서 Pytorch를 먼저 설치하여 컴퓨터에 적합한 CUDA 버전을 선택하는 것입니다.
일단 Pytorch가 설치된 경우 사용 만하면됩니다.
pip install -U “pytorch_tabular[extra]”추가 종속성 (가중치 및 바이어스 및 플롯)으로 전체 라이브러리를 설치합니다.
그리고 :
pip install -U “pytorch_tabular”필수 필수 요소.
pytorch_tabular의 소스는 Github repo _에서 다운로드 할 수 있습니다.
공개 저장소를 복제 할 수 있습니다.
git clone git://github.com/manujosephv/pytorch_tabular소스 사본이 있으면 다음과 같이 설치할 수 있습니다.
cd pytorch_tabular && pip install .[extra]튜토리얼에 대한 전체 문서를 보려면 Readthedocs를 방문하십시오
반 감독 학습
새로운 모델을 구현하려면 새로운 모델 구현 방법 자습서를 참조하십시오. 기본 및 고급 아키텍처를 다룹니다.
from pytorch_tabular import TabularModel
from pytorch_tabular . models import CategoryEmbeddingModelConfig
from pytorch_tabular . config import (
DataConfig ,
OptimizerConfig ,
TrainerConfig ,
ExperimentConfig ,
)
data_config = DataConfig (
target = [
"target"
], # target should always be a list.
continuous_cols = num_col_names ,
categorical_cols = cat_col_names ,
)
trainer_config = TrainerConfig (
auto_lr_find = True , # Runs the LRFinder to automatically derive a learning rate
batch_size = 1024 ,
max_epochs = 100 ,
)
optimizer_config = OptimizerConfig ()
model_config = CategoryEmbeddingModelConfig (
task = "classification" ,
layers = "1024-512-512" , # Number of nodes in each layer
activation = "LeakyReLU" , # Activation between each layers
learning_rate = 1e-3 ,
)
tabular_model = TabularModel (
data_config = data_config ,
model_config = model_config ,
optimizer_config = optimizer_config ,
trainer_config = trainer_config ,
)
tabular_model . fit ( train = train , validation = val )
result = tabular_model . evaluate ( test )
pred_df = tabular_model . predict ( test )
tabular_model . save_model ( "examples/basic" )
loaded_model = TabularModel . load_model ( "examples/basic" )마누 조셉 | Jirka Borovec | Jinu Sunil | Programador 인공 | 소렌 맥베스 | Chris Fonnesbeck |
Snehil Chatterjee | 널 | Abhishar Sinha | 안드레아스 | Charitarth Chugh | 이어리 |
널 | 쿠샤샤와 라비 슈리 얄리 | Luca Actis Grosso | 스털링 G. 베어드 | Teck Meng | Yinyu Nie |
Yonybresler | Liu Zhen |
과학적 간행물에 Pytorch Tabular를 사용하는 경우 게시 된 소프트웨어 및 다음 논문에 대한 인용에 감사드립니다.
@misc{joseph2021pytorch,
title={PyTorch Tabular: A Framework for Deep Learning with Tabular Data},
author={Manu Joseph},
year={2021},
eprint={2104.13638},
archivePrefix={arXiv},
primaryClass={cs.LG}
}
@software{manu_joseph_2023_7554473,
author = {Manu Joseph and
Jinu Sunil and
Jiri Borovec and
Chris Fonnesbeck and
jxtrbtk and
Andreas and
JulianRein and
Kushashwa Ravi Shrimali and
Luca Actis Grosso and
Sterling G. Baird and
Yinyu Nie},
title = {manujosephv/pytorch_tabular: v1.0.1},
month = jan,
year = 2023,
publisher = {Zenodo},
version = {v1.0.1},
doi = {10.5281/zenodo.7554473},
url = {https://doi.org/10.5281/zenodo.7554473}
}