

Pytorch tabellarische Ziele zielen darauf ab, Deep-Lernen mit tabellarischen Daten einfach und zugänglich für Fälle in der realen Welt und die Forschung zu machen. Die Kernprinzipien hinter dem Design der Bibliothek sind:
Es wurde auf den Schultern von Riesen wie Pytorch (offensichtlich) und Pytorch Lightning gebaut.
Obwohl die Installation Pytorch enthält, besteht die beste und empfohlene Möglichkeit darin, zuerst Pytorch von hier aus zu installieren und die richtige CUDA -Version für Ihren Computer aufzunehmen.
Einmal haben Sie Pytorch installiert. Verwenden Sie einfach:
pip install -U “pytorch_tabular[extra]”So installieren Sie die vollständige Bibliothek mit zusätzlichen Abhängigkeiten (Gewichte & Vorurteile & Plotly).
Und :
pip install -U “pytorch_tabular”für das nötige Nötigste.
Die Quellen für pytorch_tabular können aus dem Github repo _ heruntergeladen werden.
Sie können entweder das öffentliche Repository klonen:
git clone git://github.com/manujosephv/pytorch_tabularSobald Sie eine Kopie der Quelle haben, können Sie sie mit:
cd pytorch_tabular && pip install .[extra]Für vollständige Dokumentation mit Tutorials finden Sie Readthedocs
Semi-betriebliches Lernen
Um neue Modelle zu implementieren, finden Sie in der Tutorial zum Implementieren neuer Modelle. Es deckt sowohl grundlegende als auch fortschrittliche Architekturen ab.
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" )Manu Joseph | Jirka Borovec | Jinu Sunil | Programmador künstlich | Soren Macbeth | Chris Fonnesbeck |
Snehil Chatterjee | NULL | Abhishar Sinha | Andreas | Charitarth Chugh | Earlee |
NULL | Kushashwa Ravi Shrimali | Luca Actis Grosso | Sterling G. Baird | Teck Meng | Yinyu Nie |
Yonybresler | Liu Zhen |
Wenn Sie Pytorch Tabar für eine wissenschaftliche Veröffentlichung verwenden, würden wir uns Zitate für die veröffentlichte Software und das folgende Papier freuen:
@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}
}