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!

Build seamlessly, securely, and flexibly with MongoDB Atlas. Try free.

Build seamlessly, securely, and flexibly with MongoDB Atlas. Try free.

MongoDB Atlas lets you build and run modern apps in 125+ regions across AWS, Azure, and Google Cloud. Multi-cloud clusters distribute data seamlessly and auto-failover between providers for high availability and flexibility. Start free!

Learn More

Top comments (0)

AWS Q Developer image

Build your favorite retro game with Amazon Q Developer CLI in the Challenge & win a T-shirt!

Feeling nostalgic? Build Games Challenge is your chance to recreate your favorite retro arcade style game using Amazon Q Developer’s agentic coding experience in the command line interface, Q Developer CLI.

Participate Now

👋 Kindness is contagious

Explore this insightful piece, celebrated by the caring DEV Community. Programmers from all walks of life are invited to contribute and expand our shared wisdom.

A simple "thank you" can make someone’s day—leave your kudos in the comments below!

On DEV, spreading knowledge paves the way and fortifies our camaraderie. Found this helpful? A brief note of appreciation to the author truly matters.

Let’s Go!