1. Learn
    2. /
  2. Courses
    3. /
  3. Introduction to Python & Machine Learning (with Analytics Vidhya Hackathons)

Exercise

Train model and do prediction using Decision Tree

Let’s make first Decision Tree model. Similar to Logistic regression, we first select the input features, train our model and finally perform prediction on test data set.

Ok! time for you to build your first Decision Tree model! The pre processed trainmodified and testmodifed data are available in your workspace.

Instructions

100 XP
  • Store input variable in list "predictors"
  • Create a object of DecisionTreeClassifier
  • Do prediction for test data set
  • Export test prediction to csv file