Commit 0532f716 authored by igraf's avatar igraf
Browse files

Update README.md

parent ffb7cbda
Loading
Loading
Loading
Loading
+3 −3
Original line number Diff line number Diff line
@@ -460,9 +460,9 @@ We see that for all settings, a higher number of estimators leads to better resu

<details> <summary> This did indeed lead to better results: 150 estimators: accuracy of <b>0.481</b></summary> 

| Resized | Features | Accuracy (Dev) | Best Parameters | Comments |
| ------- | -------- | -------- | --------------- | ---- |
| 50x50   | HSV only | 0.481| `{'max_depth': 40, 'max_features': 'sqrt', 'min_samples_leaf': 2, 'min_samples_split': 2, 'n_estimators': 150}` | Time for optimization: 251 min |
| Resized | Features | Accuracy (Dev) | Best Parameters | 
| ------- | -------- | -------- | --------------- | 
| 50x50   | HSV only | 0.481| `{'max_depth': 40, 'max_features': 'sqrt', 'min_samples_leaf': 2, 'min_samples_split': 2, 'n_estimators': 150}` | 


<img  align="center" src="figures/random_forest/grid_search_results_rforest_50x50_hsv_max_depth_n_estimators.png" alt= "Grid" width="80%" height="auto">