PyCaret + SKORCH: Build PyTorch Neural Networks using Minimal Code

A low-code guide to build PyTorch Neural Networks with Pycaret

Pranay Modukuru
Towards Data Science

Photo by Danist Soh on Unsplash

Almost in every machine learning project, we train and evaluate multiple machine learning models. This often involves writing multiple lines of imports, many function calls, print statements to train individual models and compare the results across the models. The code becomes a mess when comparing different models with cross-validation loops or ensembling the models. Over time, it gets even messier when we move from classification models to regression models or vice-versa. We end up copying snippets of code from one place to another, creating chaos! We can easily avoid this chaos by just importing PyCaret!

PyCaret is a low-code machine library that allows you to create, train, and test ML models via a unified API given a regression or classification problem. PyCaret also offers various steps in a machine learning project, from data preparation to model deployment with a minimal amount of code. It can work with any model/library that follows a Scikit-Learn API such as Scikit-Learn Models, Xgboost, LightGBM, CatBoost, etc. All in all, the library is an easy-to-use productivity booster that enables fast experiments and helps you focus more on the business problem at hand.

Create an account to read the full story.

The author made this story available to Medium members only.
If you’re new to Medium, create a new account to read this story on us.

Or, continue in mobile web

Already have an account? Sign in

Responses (3)

What are your thoughts?

Looks very interesting indeed, though for a beginner like me , bit daunting. Will try to implement on the stocks time series ( it's very volatile these days) and see whether vis-a-vis pure Pycaret , your implementation catches the volatility better!
Any suggestions!
Rgds.

2

Amazing article, congrats!

Hi,
I have a problem with prediction model.