From 80c46c292d53d842dc87eff735ff463655886ddb Mon Sep 17 00:00:00 2001
From: Diego Colombatto <60887219+Colombatto@users.noreply.github.com>
Date: Thu, 20 May 2021 22:59:04 +0200
Subject: [PATCH] Updated numFolds

updated numFolds to have better Random Forest performance
---
 2_Strategies/Strategy_WhatIfScenarios.ipynb | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/2_Strategies/Strategy_WhatIfScenarios.ipynb b/2_Strategies/Strategy_WhatIfScenarios.ipynb
index 6834f9c..f62d482 100644
--- a/2_Strategies/Strategy_WhatIfScenarios.ipynb
+++ b/2_Strategies/Strategy_WhatIfScenarios.ipynb
@@ -271,9 +271,9 @@
     "pipeline = Pipeline(stages=[assembler, rf])\n",
     "\n",
     "#hyperparameters values: numTrees start, numTrees stop, numTrees num, maxDepth start, maxDepth stop, maxDepth num, numFolds\n",
-    "set1=[ 5, 25, 3, 5, 10, 3, 2]\n",
-    "set2=[ 10, 50, 3, 5, 10, 3, 3]\n",
-    "set3=[ 5, 25, 3, 5, 10, 3, 4]\n",
+    "set1=[ 5, 25, 3, 5, 10, 3, 9]\n",
+    "set2=[ 10, 50, 3, 5, 10, 3, 9]\n",
+    "set3=[ 5, 25, 3, 5, 10, 3, 27]\n",
     "\n",
     "params=[set1, set2, set3]\n",
     "models=[]\n",