This Python project implements an autism detection model using LightGBM and scikit-learn libraries. The model is trained on a dataset comprising images of children, including both autistic and non-autistic cases.
autism_detection_model.py
: Main script containing the code for training and evaluating the model.data/
: Directory containing the dataset of children's images.requirements.txt
: File listing all the required Python packages for running the code.LICENSE
: MIT License for the project.
- Clone the repository to your local machine.
- Install the required dependencies using
pip install -r requirements.txt
. - Place your dataset of children's images in the
data/
directory. - Run
autism_detection_model.py
to train and evaluate the model. - Customize the model parameters and preprocessing steps as needed.
- Python 3.x
- LightGBM
- scikit-learn
- PIL (Python Imaging Library)
The dataset used for training the model is not included in this repository due to privacy and licensing constraints. Please ensure you have appropriate permissions to use and distribute your dataset.
This project is licensed under the MIT License.
Pavan Sriram Kannuri