Skip to content

Hemanthpolineni/Autism-Prediction

Folders and files

NameName
Last commit message
Last commit date

Latest commit

Β 

History

9 Commits
Β 
Β 
Β 
Β 
Β 
Β 

Repository files navigation

🧠 Autism Dataset β€” Preprocessing, Modeling & Deployment Pipeline

This repository contains a complete, reproducible end-to-end machine learning pipeline for the Autism Spectrum Disorder (ASD) Screening Dataset for Children, covering data preprocessing, exploratory data analysis (EDA), feature engineering, model training, evaluation, and deployment.

The project is implemented using a Jupyter Notebook for experimentation and analysis, and the trained model is deployed using Flask as a lightweight web application.


πŸ“Œ Features

πŸ”§ Data Preprocessing

  • Handling missing values
  • Encoding categorical variables
  • Scaling numerical features
  • Outlier detection and removal
  • Data leakage prevention using a configurable DROP_LEAKAGE flag
  • Finding top 10 features

πŸ€– Machine Learning Pipeline

  • Train/test split
  • Classical ML models:
    • Decision Tree Classifer
    • Other baseline classifiers
  • Model evaluation using:
    • Accuracy
    • Precision
    • Recall
    • F1-score
  • Model & preprocessing pipeline serialization

🌐 Model Deployment (Flask)

  • Trained ML model deployed using Flask
  • Reuse of saved preprocessing pipeline in production
  • Modular and extensible backend structure

πŸ“ Project Structure

autism-pipeline/
β”‚
β”œβ”€β”€ autism.csv                       # Cleaned dataset
β”œβ”€β”€ Autism-Child-Data.arff           # Original ARFF dataset
β”‚
β”œβ”€β”€ models.pkl                       # PKL file 
β”œβ”€β”€ preprocessor.pkl                 # PKL file
β”‚
β”œβ”€β”€ autism_pipeline_notebook.ipynb   # EDA + preprocessing + training
β”‚                          
β”œβ”€β”€ templates/
β”‚   └── index.html                   # Fronted
β”‚
β”œβ”€β”€ app.py                           # Flask application
β”œβ”€β”€ train.py                         # Model training & serialization
β”œβ”€β”€ README.md                        # Project documentation
└── .gitignore


πŸš€ Clone the Repository

git clone https://github.com/your-username/autism-pipeline.git
cd autism-pipeline

About

No description, website, or topics provided.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published