A Streamlit-based Machine Learning platform that allows users to view data, clean data, visualize it, train models, compare two models, and evaluate performance. Users can also perform predictions on new datasets, save their models, and reuse them for future predictions. This project is designed to simulate a real-world ML workflow.
🔗 https://train-your-own-model.streamlit.app/
- Linear Regression
- Binary Classification
- Multiclass Classification (Shallow NN)
- K-Means
- Anomaly Detection (Gaussian)
- Decision Tree
- Linear Regression
- Binary Classification
- Multiclass Classification (Shallow NN)
- K-Means
- Decision Tree
- Random Forest
- XGBoost
- Data Preview: View your dataset and a statistical summary.
- Cleaning: Clean your dataset by removing duplicates, encoding categorical values, filling missing values, etc. Includes a feature for feature transformation.
- Visualization: Visualize your dataset in four forms: scatterplot, histogram, boxplot, and heatmap.
- Standardization: Standardize your dataset for better model performance.
- Training: Select the Algorithm Type, Model Type, and Implementation Type. Set parameters to train models and compare two different models simultaneously.
- Testing: Evaluate your model performance on the
X_testset. - Prediction: Predict values for a new dataset based on the trained model and download the results.
- Model Selection: Choose from your list of saved models.
- Activation: Activate the selected model for use.
- Prediction: Predict results based on the activated model.
- Export: Download the predicted data.
- Upload the dataset you want to use.
- Clean the dataset.
- Apply transformations (Log, Square, Square Root, or Absolute) to make data distributions more symmetric.
- Select the features for X and y.
- Split the dataset into Train/Test sets or Train/CV/Test sets.
- Standardize the data splits.
- Select the model you wish to train.
- (Optional) Compare two models by clicking Compare Models.
- View results based on the CV or Train dataset.
- Evaluate final performance on the Test dataset.
- Run predictions on new data and download the output.
- Select the saved model you want to use.
- Activate it.
- Navigate to the Prediction tab and upload the data you wish to predict.
NOTE: In Use Existing Model, there is a specific model format requirement. Currently, you can only use existing models that were created and saved within this app.
Ensure you have Python 3.x installed on your machine.
git clone [https://github.com/BEGINNERUSER-git/Train-your-own-Model.git](https://github.com/BEGINNERUSER-git/Train-your-own-Model.git)
cd Train-your-own-Modelpip install -r requirements.txtstreamlit run app.pyTrain.your.model.1.1.mp4
Copyright © 2026 BEGINNERUSER-git
This project is licensed under the MIT License – see the LICENSE file for details.