Update README.md
Browse files
README.md
CHANGED
@@ -21,15 +21,18 @@ The model analyzed in this card estimates someone's age. This project has been d
|
|
21 |
|
22 |
To determine the performance of the model, the following metrics have been used:
|
23 |
- MSE, this metric measures how close the regression line is to the data points.
|
24 |
-
<br>   - *Our model's MSE:*
|
25 |
- RMSE, this metric measures the mean error that can be made.
|
26 |
-
<br>   - *Our model's RMSE:*
|
27 |
- MAE, this is a measure for model accuracy. The MAE is the average error that the model's predictions have in comparison with their corresponding actual targets.
|
28 |
-
<br>   - *Our model's MAE:*
|
29 |
|
30 |
Ideally, the RMSE and the MAE should be close to each other. When there is a big difference in these two numbers, it is an indication of variance in the individually errors.
|
31 |
|
32 |
-
Our results show that the prediction model can be
|
|
|
|
|
|
|
33 |
|
34 |
## Limitations
|
35 |
- **Lighting**
|
@@ -46,9 +49,9 @@ Train data: 70%
|
|
46 |
Test data: 30%
|
47 |
|
48 |
Our model has been made by trial and error. The following architecture is the outcome:
|
49 |
-
- Hidden layers:
|
50 |
-
- Batch size
|
51 |
-
- Epochs:
|
52 |
-
- Optimizer:
|
53 |
-
- Activation:
|
54 |
|
|
|
21 |
|
22 |
To determine the performance of the model, the following metrics have been used:
|
23 |
- MSE, this metric measures how close the regression line is to the data points.
|
24 |
+
<br>   - *Our model's MSE:* 60.9
|
25 |
- RMSE, this metric measures the mean error that can be made.
|
26 |
+
<br>   - *Our model's RMSE:* 7.8
|
27 |
- MAE, this is a measure for model accuracy. The MAE is the average error that the model's predictions have in comparison with their corresponding actual targets.
|
28 |
+
<br>   - *Our model's MAE:* 5.2
|
29 |
|
30 |
Ideally, the RMSE and the MAE should be close to each other. When there is a big difference in these two numbers, it is an indication of variance in the individually errors.
|
31 |
|
32 |
+
Our results show that the prediction model can be around 8 years off of the actual age of a person.
|
33 |
+
|
34 |
+
We also looked at how the model performs in different age, gender and race classes. It seemed the model predicted the ages of people between 20 and 30 better than the rest. The model could also predict the ages of females better than males. The race that the model can predict the best is East Asian.
|
35 |
+
|
36 |
|
37 |
## Limitations
|
38 |
- **Lighting**
|
|
|
49 |
Test data: 30%
|
50 |
|
51 |
Our model has been made by trial and error. The following architecture is the outcome:
|
52 |
+
- Hidden layers: 7
|
53 |
+
- Batch size: 128
|
54 |
+
- Epochs: 65
|
55 |
+
- Optimizer: adam
|
56 |
+
- Activation: ReLu & Linear
|
57 |
|