An open API service indexing awesome lists of open source software.

https://github.com/haseeeb21/train-your-machine-learning-model

Train your Machine Learning model related to Classification or Regression task. Simply upload your csv or excel file to the website, select the target column and the model to train. Save the model in your required format Joblib or Pickle.
https://github.com/haseeeb21/train-your-machine-learning-model

Last synced: about 1 month ago
JSON representation

Train your Machine Learning model related to Classification or Regression task. Simply upload your csv or excel file to the website, select the target column and the model to train. Save the model in your required format Joblib or Pickle.

Awesome Lists containing this project

README

          

# Train-Your-Machine-Learning-Model
Train your Machine Learning model related to Classification or Regression task. Simply upload your csv or excel file to the website, select the target column and the model to train. Save the model in your required format Joblib or Pickle.

## Web App URL

https://train-your-machine-learning-model.streamlit.app/

Few datasets are also uploaded to try out the webapp :)

## Note That

The *csv* or *xlsx* files columns / features should be in Label Encoding format to train the model.

## Classification

Scikit-learn Classification models available to train are,

- K Nearest Neighbor (KNN) - K Neighbor Classifier
- Support Vector Machine (SVM) - Support Vector Classification (SVC)
- Logistic Regression (LR)
- Random Forest (RF) - Random Forest Classifier
- Decision Tree - Decision Tree Classifier
- Multi-layer Perceptron (MLP) - Multi-layer Perceptron Classifier
- Stochastic Gradient Descent (SGD) - Stochastic Gradient Descent Classifier

## Regression

Scikit-learn Regression models available to train are,

- Linear Regression
- Ridge Regression
- Lasso Regression
- Elastic Net
- Decision Tree - Decision Tree Regressor
- Random Forest - Random Forest Regressor
- Gradient Boosting - Gradient Boosting Regressor
- Support Vector Machine (SVM) - Support Vector Regression (SVR)
- K Nearest Neighbor (KNN) - K Neighbors Regressor
- Multi-layer Perceptron (MLP) - MLP Regressor

### Saving format

You can either save the trained model in `.joblib` or `.pkl` format.