Merge pull request #23 from zhi-bao/prune_inference

Fix the bug in pruning tree predictions.

Author: Eleven1Liu

libmultilabel/linear/tree.py

@@ -101,7 +101,7 @@ def _separate_model_for_pruning_tree(self):
         self.subtree_models = []
         for i in range(len(self.root.children)):
             subtree_weights_start = self.node_ptr[self.root.children[i].index]
-            subtree_weights_end = self.node_ptr[self.root.children[i+1].index] if i+1 < len(self.root.children) else -1
+            subtree_weights_end = self.node_ptr[self.root.children[i+1].index] if i+1 < len(self.root.children) else self.node_ptr[-1]
             slice = np.s_[:, subtree_weights_start:subtree_weights_end]
             subtree_flatmodel = linear.FlatModel(
                 name="subtree-flattened-tree",