Remove name of TreeModel since we only have one now.

Author: Eleven1Liu

libmultilabel/linear/tree.py

@@ -38,12 +38,11 @@ def dfs(self, visit: Callable[[Node], None]):
 
 class TreeModel:
     def __init__(self,
-                 name: str,
                  root: Node,
                  flat_model: linear.FlatModel,
                  weight_map: np.ndarray,
                  ):
-        self.name = name
+        self.name = 'tree'
         self.root = root
         self.flat_model = flat_model
         self.weight_map = weight_map
@@ -151,7 +150,7 @@ def visit(node):
     pbar.close()
 
     flat_model, weight_map = _flatten_model(root)
-    return TreeModel('tree', root, flat_model, weight_map)
+    return TreeModel(root, flat_model, weight_map)
 
 
 def _build_tree(label_representation: sparse.csr_matrix,