|
a |
|
b/README.md |
|
|
1 |
# Obesity Risk Prediction CNN Model |
|
|
2 |
|
|
|
3 |
This repository hosts a Convolutional Neural Network (CNN) model that predicts the risk of obesity in individuals with an impressive accuracy of 87.01%. Developed on Google Colab, this model harnesses the platform's powerful GPUs and collaborative environment. |
|
|
4 |
|
|
|
5 |
## Model Overview |
|
|
6 |
|
|
|
7 |
- *Objective*: Accurately predict obesity risk from input data. |
|
|
8 |
- *Accuracy*: Attained 87.01% on the test dataset. |
|
|
9 |
- *Dataset*: Sourced from a Kaggle competition via its API. |
|
|
10 |
- *Frameworks*: Built with TensorFlow/Keras. |
|
|
11 |
- *Training Environment*: Google Colab. |
|
|
12 |
|
|
|
13 |
## Features |
|
|
14 |
|
|
|
15 |
- *Input*: Accepts data with shape (22, 1), including normalization and dropout layers. |
|
|
16 |
- *Output*: Outputs the probability of obesity risk. |
|
|
17 |
- *Architecture*: Consists of 5 layers, featuring convolutional layers, dropout layers, batch normalization, and fully connected layers. |
|
|
18 |
- *Optimization*: Utilizes the Adam optimizer with a learning rate of 0.00005. |
|
|
19 |
|
|
|
20 |
## Results |
|
|
21 |
|
|
|
22 |
In a sample test of 10, the model correctly predicted 8 instances, indicating potential areas for improvement to enhance accuracy. |
|
|
23 |
|
|
|
24 |
--- |
|
|
25 |
|
|
|
26 |
Your contributions and feedback are invaluable as we continue to refine this model. Feel free to clone and explore its capabilities! |