Skip to main content
`Ctrl`+`K`

# 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