From 7e3b80cc1da5d407cd2a0025ea77aa9c4d25812a Mon Sep 17 00:00:00 2001
From: Pedro Silva <psilva+git@pedrosilva.pt>
Date: Sun, 27 Oct 2013 22:45:55 +0100
Subject: [PATCH] WIP: add classification task

---
 beetsplug/learn.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/beetsplug/learn.py b/beetsplug/learn.py
index 9fac231f7..1dd4d60af 100644
--- a/beetsplug/learn.py
+++ b/beetsplug/learn.py
@@ -101,6 +101,7 @@ class LearnPlugin(beets.plugins.BeetsPlugin):
 
         self.config.add({'attributes': []})
         self.config.add({'clusters': 2})
+        self.config.add({'classify': []})
         self.config.add({'format': ''})
         self.config.add({'kind': 'numeric'})
         self.config.add({'plot': False})
@@ -121,6 +122,12 @@ class LearnPlugin(beets.plugins.BeetsPlugin):
                                         type=int,
                                         help='how many clusters to find')
 
+        self._command.parser.add_option('-C', '--classify',
+                                        action='callback', dest='classify',
+                                        metavar='LIST',
+                                        callback=beets.ui.vararg_callback,
+                                        help='list of labels to classify')
+
         self._command.parser.add_option('-f', '--format',
                                         action='store', type=str,
                                         help='print with custom format',
@@ -162,10 +169,12 @@ class LearnPlugin(beets.plugins.BeetsPlugin):
             features = self.config['attributes'].get(list)
             fmt = self.config['format'].get(str)
             k = self.config['clusters'].get(int)
+            targets = self.config['classify'].get(list)
             plot = self.config['plot'].get(bool)
             savefig = self.config['savefig'].get(str)
             test = self.config['test'].get(list)
             train = self.config['train'].get(list)
+
             if not fmt:
                 fmt = '$albumartist - $album - $title'
             fmt += ' - {0}'