grep train and predict as key words in the classifiers
fix model stage checks
Merge pull request #20 from robertzk/fix_model_stage

Fix model stage to allow flexible naming train and predict functions in the lib/classifier