add basic framework for classification model training