From a96bcf9e8d1304639bf0db8e8c23a767e6ee5754 Mon Sep 17 00:00:00 2001
From: Nikolai Hartmann <Nikolai.Hartmann@physik.uni-muenchen.de>
Date: Mon, 10 Sep 2018 13:26:31 +0200
Subject: [PATCH] fixing kfold validation
---
toolkit.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/toolkit.py b/toolkit.py
index cc479a4..3d52a59 100755
--- a/toolkit.py
+++ b/toolkit.py
@@ -840,7 +840,7 @@ class ClassificationProject(object):
if self._train_val_idx is None:
if self.kfold_splits is not None:
kfold = KFold(self.kfold_splits, shuffle=True, random_state=self.shuffle_seed)
- for i, train_val_idx in kfold.split(self.x_train):
+ for i, train_val_idx in enumerate(kfold.split(self.x_train)):
if i == self.kfold_index:
self._train_val_idx = train_val_idx
break
--
GitLab