seperate train and classify