trained sucessfully--- next is to classify the actual set