curdrice-v2 0.1

A ML model training automation tool

ML Workflow Automation Tool

The main goal is to simplify the machine learning workflow for users with different levels of expertise, through a CLI app. We ask the user for certain inputs like the dataset file or folder, and the type of ML problem and target variable (if necessary). We then perform standard data preprocessing (not dataset specific) and feature selection (if necessary), and train relevant models for the data and present a comparative analysis of the models trained, along with downloadable model weight files in the joblib format.

Key Features

• Compare various machine learning algorithms
• Performance comparison
• Model training result comparison through CLI
• All model files stored as a zip of joblib files

Preview

Deployment

Find out which ML model suits your needs by installing our package from pypi

pip install curdrice

Usage

Use the following command for help -

curdrice --help

Template code

curdrice -p <db_path> -d <zip_name> <type> <target>

Acknowledgements

• Numpy
• Pandas
• Openpyxl
• Scikit Learn
• Scikit Learn Extra
• Joblib

Authors

@hrishitb - @pranayobla - @shriharik - @ankitthomas