adding dimensional selection in set_classify and predict functions