cog

cog train: bring functionality on par with cog predict

#1391

Merged