search

JamesOwers / midi_degradation_toolkit

A toolkit for generating datasets of midi files which have been degraded to be 'un-musical'.
MIT License
38 stars 5 forks source link

Weights #72

Closed apmcleod closed 5 years ago

apmcleod commented 5 years ago

This adds a --weight option to train_tasks.py, which passes a weight for each class to the optimizer, inversely proportional to that class's count in the train set.

This also renames eval_script to eval_task (like train_task)