Fix cost function (!!!)
Add --min-count parameter for final train phase