A machine learning project to predict autism spectrum disorder based on input features. This repository contains data, notebooks, trained models, and supporting scripts for building, evaluating, and deploying the classifier.
Autism-Detection/
βββ train.csv
βββ test.csv
βββ AutismDetection.ipynb
βββ best_model.pkl
βββ encoders.pkl
βββ README.mdThis project aims to develop a machine learning model that can detect autism (or likelihood thereof) given input variables (demographic, clinical, behavioral features). The pipeline involves:
- Data loading & exploration
- Data preprocessing / feature engineering
- Model training & hyperparameter tuning
- Evaluation (accuracy, recall, precision, confusion matrix)
- Serialization of the best model + encoders
- Optionally, inference / deployment
- Python 3.7+
- Required libraries
numpy pandas scikit-learn matplotlib / seaborn joblib / pickle jupyter
# Clone the repo
git clone https://github.com/mehershiri/Autism-Detection.git
cd Autism-Detection
# Install dependecies if needed
pip install -r requirements.txt
# Start Jupyter Lab/ Notebook using the following commands:
jupyter notebook
or
jupyter lab
# Open and run AutismDetection.ipynb step by step.
-The notebook loads train.csv and test.csv
-It preprocesses features, encodes categorical variables
-Trains multiple models, chooses the best one
-Saves the best model & encoders (best_model.pkl, encoders.pkl)
-Evaluates performance on the test set