Skip to content

Commit 07b829e

Browse files
committed
change the store format of weights of the node to csc_matrix
1 parent cb651c1 commit 07b829e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

libmultilabel/linear/tree.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -212,7 +212,7 @@ def _train_node(y: sparse.csr_matrix, x: sparse.csr_matrix, options: str, node:
212212
meta_y = sparse.csr_matrix(np.hstack(meta_y))
213213
node.model = linear.train_1vsrest(meta_y, x, options, False)
214214

215-
node.model.weights = sparse.csr_matrix(node.model.weights)
215+
node.model.weights = sparse.csc_matrix(node.model.weights)
216216

217217

218218
def _flatten_model(root: Node) -> tuple[linear.FlatModel, np.ndarray]:

0 commit comments

Comments
 (0)