Skip to content

aliasvishnu/DecisionTree-Visualization

Repository files navigation

Decision Tree And Visualization

Following modules used

  • DecisionTreeClassifier (iris.py)
  • DecisionTreeRegressor
  • RandomForestClassifier (iris.py)
  • ExtraTreesRegressor (facecompletion.py)
  • LinearRegression (facecompletion.py)

Problems

  • Iris classification problem - 3 way classification
  • Face completion - Complete bottom half of face given upper half

Just run python filename.py to get output.

Here is the output from the facecompletion program.

Discussion

Face Completion

We can see the kNN has just picked up the bottom half of another person whose upper face matched with our test subject. Extra trees have produced an averaging of results from each tree, thus giving a blurry look to it whereas Linear regression has produced noisy, rough looking images.

Further Readings