AutoML / README.md
Gourav18's picture
Update README.md
2d8bdaf verified
metadata
license: mit
language:
  - en
metrics:
  - accuracy
  - roc_auc
  - precision
  - f1
  - confusion_matrix
base_model:
  - Gourav18/AutoML
new_version: Gourav18/AutoML
library_name: sklearn
tags:
  - code
  - AutoMl

AutoML

AutoML Application

An automated machine learning application built with Streamlit that helps users to:

  • Load and preprocess data
  • Train multiple ML models (Logistic Regression, SVM, Random Forest)
  • Automatically select the best model
  • Visualize model performance
  • Make predictions with the trained model

Features

  • Data Upload & Analysis: Upload CSV or Excel files, view statistics, and visualize correlations
  • Automated Data Preprocessing: Handle missing values and categorical encoding
  • Model Training: Train multiple models with hyperparameter tuning
  • Model Visualization: Compare model performance with various metrics
  • Prediction: Make predictions using the trained model

Setup

  1. Clone this repository
  2. Install the requirements:
    pip install -r requirements.txt
    
  3. Run the application:
    streamlit run main.py