Decision Tree

In this project we solve the banknote classification task with a decision tree.

Load Data Set

Task: Load the banknotes data set (cf. Detecting Forgery with k-NN). Drop the entropy column.

# your solution

Decision Tree

Task: Train and evaluate a decision tree for banknote classification. Use cost-complexity pruning with corresponding parameter chosen by hyperparameter optimization.

Solution:

# your solution

Decision Surface

Task: Train a second model based on variance and skewness only. Plot the decision surface (the surface separating the classes) with Matplotlib’s contour or contourf.

Solution:

# your solution