Merge pull request #362 from bsc-wdc/rf-regressor-classification-fix

RF classification fix
bsc-wdc · Oct 29, 2021 · 69cc801 · 69cc801
2 parents b2f4704 + 66e9371
commit 69cc801
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/Jenkinsfile b/Jenkinsfile
@@ -11,7 +11,7 @@ def setGithubCommitStatus(state, description) {
 
 pipeline {
     options {
-        timeout(time: 4, unit: 'HOURS')
+        timeout(time: 5, unit: 'HOURS')
     }
     agent {
         node {

diff --git a/dislib/trees/decision_tree.py b/dislib/trees/decision_tree.py
@@ -148,7 +148,8 @@ def predict(self, x_row):
             )
             branch_predictions.append(pred)
         return _merge_branches(
-            None, *branch_predictions, classification=self.n_classes is None
+            None, *branch_predictions,
+            classification=self.n_classes is not None
         )