fix the bug

ASUS-AICS · Aug 16, 2024 · 56bf1a5 · 56bf1a5
1 parent 0b638b0
commit 56bf1a5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/libmultilabel/linear/tree.py b/libmultilabel/linear/tree.py
@@ -141,7 +141,7 @@ def train_tree(
     def count(node):
         nonlocal num_nodes
         num_nodes += 1
-        node.num_nnz_feat = np.count_nonzero(label_feature_used[:, node.label_map].sum(axis=0))
+        node.num_nnz_feat = np.count_nonzero(label_feature_used[:, node.label_map].sum(axis=1))
 
     root.dfs(count)