From a548e81e98ba3e151ded6ed94c6874e0179ae0af Mon Sep 17 00:00:00 2001
From: Nikolai Hartmann <Nikolai.Hartmann@physik.uni-muenchen.de>
Date: Wed, 29 Aug 2018 13:09:34 +0200
Subject: [PATCH] Raise ValueError when w_train_tot requested, but data not
loadad
---
toolkit.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/toolkit.py b/toolkit.py
index d0f44ce..a6b3a0c 100755
--- a/toolkit.py
+++ b/toolkit.py
@@ -823,7 +823,7 @@ class ClassificationProject(object):
else:
class_weight = self.balanced_class_weight
if not self.data_loaded:
- self._w_train_tot = None
+ raise ValueError("Data not loaded! can't calculate total weight")
if self._w_train_tot is None:
if self.apply_class_weight:
self._w_train_tot = self.w_train*np.array(class_weight)[self.y_train.astype(int)]
--
GitLab