This repository contains a prototype machine learning model developed by the Mathemedics 2000 team. The model leverages linear regression to analyze the Pima Indian diabetes dataset provided by the Indian government. The primary function of this model is to predict whether a patient has diabetes based on their medical report, using binary comparisons.
- Linear Regression Model: Utilizes linear regression for prediction.
- Binary Comparison: Provides a binary output indicating diabetes presence.
- Pima Indian Diabetes Dataset: Uses a well-known dataset from the Indian government.
Ensure you have the following installed:
- Python 3.7 or higher
- Git
-
Clone the repository
git clone https://github.com/jhaabhijeet864/Prototype_Diabetes_Model.git cd Prototype_Diabetes_Model
-
Create a virtual environment
python -m venv venv
-
Activate the virtual environment
-
On Windows:
.\venv\Scripts\activate
-
On macOS/Linux:
source venv/bin/activate
-
-
Install the required dependencies
pip install -r requirements.txt
-
Prepare the dataset
Ensure the dataset is placed in the appropriate directory as mentioned in the dataset section.
-
Run the training script
python train_model.py
This script will train the model on the Pima Indian diabetes dataset.
-
Use the trained model
python predict.py --input <path_to_input_file>
Replace
<path_to_input_file>
with the path to the file containing the patient's medical report.
Here is an example of how to make a prediction:
python predict.py --input sample_patient_data.csv
The output will indicate whether the patient has diabetes or not based on the provided medical report.
The Pima Indian diabetes dataset can be downloaded from Kaggle. Ensure to place the dataset in the data
directory of this repository.
We welcome contributions! Please follow these steps:
- Fork the repository.
- Create a new branch (
git checkout -b feature-branch
). - Make your changes.
- Commit your changes (
git commit -m 'Add some feature'
). - Push to the branch (
git push origin feature-branch
). - Open a Pull Request.
This project is licensed under the MIT License. See the LICENSE file for details.
Feel free to customize this README further based on specific details you want to include about your project.