Python Code For Decision Tree