train on specific categories
fmt