From f3ccb2f17576541dcce202cc273dd8836ed6ef65 Mon Sep 17 00:00:00 2001
From: johannes bilk <johannes.bilk-2@exp2.physik.uni-giessen.de>
Date: Thu, 10 Aug 2023 21:56:43 +0300
Subject: [PATCH] forrest test
---
forrest-test.ipynb | 5 +++--
1 file changed, 3 insertions(+), 2 deletions(-)
diff --git a/forrest-test.ipynb b/forrest-test.ipynb
index 7487af5..5d52705 100644
--- a/forrest-test.ipynb
+++ b/forrest-test.ipynb
@@ -24,7 +24,7 @@
"from machineLearning.rf import (\n",
" RandomForest, DecisionTree,\n",
" Gini, Entropy, MAE, MSE,\n",
- " Mode, Mean, Confidence,\n",
+ " Mode, Mean, Confidence, Probabilities,\n",
" CART, ID3, C45,\n",
" AdaBoosting, GradientBoosting,\n",
" Majority, Confidence, Average, Median\n",
@@ -85,7 +85,8 @@
"# Shuffle the training data\n",
"trainIndex = np.random.permutation(len(trainData))\n",
"trainData = trainData[trainIndex]\n",
- "trainLabels = trainLabels[trainIndex]"
+ "trainLabels = trainLabels[trainIndex]\n",
+ "trainTargets = trainTargets[trainIndex]"
]
},
{
--
GitLab