Decision Tree Classifier Python Code For Prime