added option to dont train on classify call