Skip to content

Commit c93b1ed

Browse files
authored
Merge pull request #324 from ChengYehLi/master
change the storage format of node's weights to csc_matrix
2 parents 71cb589 + 07b829e commit c93b1ed

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)