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!
Top comments (0)