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%. +
The detailed procedures can be found in the following notebooks: +
[ml_xgboost.ipynb](notebooks/ml_xgboost.ipynb) +
[ml_grad_boost_tree.ipynb](notebooks/ml_grad_boost_tree.ipynb) ## Contributing