Backup_Repos/beets
1
0
Fork 0
mirror of https://github.com/beetbox/beets.git synced 2025-12-31 13:02:47 +01:00
Code Issues Projects Releases Wiki Activity

WIP: add classification task

Browse source
This commit is contained in:
Pedro Silva 2013-10-27 22:45:55 +01:00
parent 8f9e35f081
commit 7e3b80cc1d
1 changed files with 9 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
beetsplug/learn.py
View file

@ -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}'