From e087bbaafad5013d2f9a87040a98ec6e34fe968a Mon Sep 17 00:00:00 2001
From: Benjamin Paul Jaeger <benjamin.jaeger@cern.ch>
Date: Fri, 6 Nov 2020 15:41:59 -0800
Subject: [PATCH] add argument. Ref #60
---
nnfwtbn/model.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/nnfwtbn/model.py b/nnfwtbn/model.py
index a7fdb90..e9ede9a 100644
--- a/nnfwtbn/model.py
+++ b/nnfwtbn/model.py
@@ -69,7 +69,7 @@ class CrossValidator(ABC):
"""
@abstractmethod
- def select_training(self, df, fold_i):
+ def select_training(self, df, fold_i, for_predicting = False):
"""
Returns the index array to select all training events from the dataset for the
given fold.
--
GitLab