Updated numFolds

updated numFolds to have better Random Forest performance
danteGPT · May 20, 2021 · 80c46c2 · 80c46c2
1 parent 907f621
commit 80c46c2
Showing 1 changed file with 3 additions and 3 deletions.
diff --git 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",