From 26080bfada579932606d7a5b8ce87fa5272bed2b Mon Sep 17 00:00:00 2001
From: Arman Ulusoy <3016148@stud.hs-mannheim.de>
Date: Wed, 12 Jun 2024 16:03:38 +0200
Subject: [PATCH] update

---
 README.md | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/README.md b/README.md
index ddbdcc7..f37305d 100644
--- a/README.md
+++ b/README.md
@@ -202,6 +202,10 @@ The detection ability of the NeuroKit2 library is tested to detect features in t
 The exact process can be found in the notebook: [features_detection.ipynb](notebooks/features_detection.ipynb).
 
 ### ML-models
+First, the grid was tested to find the best model which was then trained to identify the best hyperparameters out of it. That way, an accuracy of 83 % was achieved with the XGBoost classifier. The Gradient Boosting Tree Classifier had an accuracy of 82%.
+<br>The detailed procedures can be found in the following notebooks: 
+<br>[ml_xgboost.ipynb](notebooks/ml_xgboost.ipynb)
+<br>[ml_grad_boost_tree.ipynb](notebooks/ml_grad_boost_tree.ipynb)
 
 
 ## Contributing