Classification Model In Python Code