From cb83fad44bb4f00c9ef46c88b2e1d180768e5118 Mon Sep 17 00:00:00 2001
From: Nikolai Hartmann <Nikolai.Hartmann@physik.uni-muenchen.de>
Date: Fri, 17 Aug 2018 11:59:32 +0200
Subject: [PATCH] turn back on division by mean weight
---
toolkit.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/toolkit.py b/toolkit.py
index e25b458..52b911f 100755
--- a/toolkit.py
+++ b/toolkit.py
@@ -775,7 +775,7 @@ class ClassificationProject(object):
class_weight = self.balanced_class_weight
if self._w_train_tot is None:
self._w_train_tot = self.w_train*np.array(class_weight)[self.y_train.astype(int)]
- #self._w_train_tot /= np.mean(self._w_train_tot)
+ self._w_train_tot /= np.mean(self._w_train_tot)
return self._w_train_tot
--
GitLab