remove support for verbosity 6, add support for no label assigned by …

…meka situation
scikit-multilearn · Feb 8, 2017 · f701d1f · f701d1f
1 parent eb478e5
commit f701d1f
Showing 1 changed file with 12 additions and 11 deletions.
diff --git a/skmultilearn/ext/meka.py b/skmultilearn/ext/meka.py
@@ -225,17 +225,18 @@ def parse_output(self):
                 '\n'))[0].split('(')[1].split('=')[1].split(')')[0]))
         self.predictions = self.output.split(predictions_split_head)[1].split(
             predictions_split_foot)[0].split('\n')[1:-1]
-        self.predictions = map(lambda z: map(lambda f: int(f.strip()),  z.split(',')), map(
-            lambda y: y.split(']')[0], map(lambda x: x.split('] [')[1], self.predictions)))
-
-        if self.verbosity == 6:
-            self.results = sparse.csr_matrix(self.predictions)
-        elif self.verbosity == 5:
-            self.results = sparse.lil_matrix(
-                (self.instance_count, self.label_count), dtype='int')
-            for row in xrange(self.instance_count):
-                for label in self.predictions[row]:
-                    self.results[row, label] = 1
+
+        self.predictions = map(lambda y: y.split(']')[0], map(lambda x: x.split('] [')[1], self.predictions))
+        self.predictions = map(lambda z: filter(lambda a: len(a) > 0, map(lambda f: f.strip(), z.split(','))), self.predictions)
+        self.predictions = map(lambda z: map(lambda a: int(a), z), self.predictions)
+
+        assert self.verbosity == 5
+
+        self.results = sparse.lil_matrix(
+            (self.instance_count, self.label_count), dtype='int')
+        for row in xrange(self.instance_count):
+            for label in self.predictions[row]:
+                self.results[row, label] = 1
 
         statistics = filter(lambda x: len(x) > 0 and '==' not in x, self.output.split(
             '== Evaluation Info')[1].split('\n'))