Merge pull request #285 from ntumlgroup/tree-name

Fixed missing parameter

Author: cjlin1

example_config/EUR-Lex/tree_l2svm.yml

@@ -0,0 +1,17 @@
+# data
+training_file: data/EUR-Lex/train.txt
+test_file: data/EUR-Lex/test.txt
+data_name: EUR-Lex
+
+# train
+seed: 1337
+linear: true
+liblinear_options: "-s 2 -B 1 -e 0.0001 -q"
+linear_technique: tree
+
+# eval
+eval_batch_size: 256
+monitor_metrics: [P@1, P@3, P@5]
+metric_threshold: 0
+
+data_format: txt

libmultilabel/linear/tree.py

@@ -256,6 +256,7 @@ def visit(node):
     root.dfs(visit)
 
     model = linear.FlatModel(
+        name='flattened-tree',
         weights=np.hstack(weights),
         bias=bias,
         thresholds=0,