DEV Community

Julien Simon
Julien Simon

Posted on • Originally published at julsimon.Medium on

Training models automatically on tabular data with Hugging Face AutoTrain

In this video, I use AutoTrain, an AutoML product designed by Hugging Face, to train a multi-class classification model on tabular data (the PetFinder dataset from Kaggle).

Original dataset: https://www.kaggle.com/competitions/petfinder-adoption-prediction/

Dataset on the hub: https://huggingface.co/datasets/juliensimon/autotrain-data-petfinder-demo

Model on the hub: https://huggingface.co/juliensimon/autotrain-tabular-demo-762523398

Notebook: https://gitlab.com/juliensimon/huggingface-demos/-/tree/main/autotrain-petfinder

New to Transformers? Check out the Hugging Face course!

Discussion (0)