Doctorado-ML/stree
1
1
Fork 0
mirror of https://github.com/Doctorado-ML/STree.git synced 2025-08-19 01:16:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix unintended nested if in partition

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2021-04-08 08:27:31 +02:00
parent 0f89b044f1
commit c36f685263
2 changed files with 10 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
stree/Strees.py
View File

@@ -460,9 +460,9 @@ class Splitter:
# in predcit time just use the column computed in train time
# is taking the classifier of class <col>
col = node.get_partition_column()
if col == -1:
# No partition is producing information gain
data = np.ones(data.shape)
if col == -1:
# No partition is producing information gain
data = np.ones(data.shape)
data = data[:, col]
self._up = data > 0