Skip to content

Commit 3aeff7b

Browse files
committed
change code form
1 parent 563c57d commit 3aeff7b

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

libmultilabel/linear/tree.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -164,10 +164,10 @@ def count(node):
164164

165165
def visit(node):
166166
if node.is_root:
167-
relevant_instances = y[:, node.label_map].getnnz(axis=1) >= 0
167+
_train_node(y, x, options, node)
168168
else:
169169
relevant_instances = y[:, node.label_map].getnnz(axis=1) > 0
170-
_train_node(y[relevant_instances], x[relevant_instances], options, node)
170+
_train_node(y[relevant_instances], x[relevant_instances], options, node)
171171
pbar.update()
172172

173173
root.dfs(visit)

0 commit comments

Comments
 (0)