LisanneH commited on
Commit
a7a4603
·
1 Parent(s): 7ea6746

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +12 -9
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> &ensp; - *Our model's MSE:*
25
  - RMSE, this metric measures the mean error that can be made.
26
- <br> &ensp; - *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> &ensp; - *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 ... years off of the actual age of a person.
 
 
 
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> &ensp; - *Our model's MSE:* 60.9
25
  - RMSE, this metric measures the mean error that can be made.
26
+ <br> &ensp; - *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> &ensp; - *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