Lab 02: Decision Tree With Scikit-Learn: About The Mushroom Data Set
Lab 02: Decision Tree With Scikit-Learn: About The Mushroom Data Set
In this assignment, you are going to build a decision tree on the Mushroom dataset, with the support
from scikit-learn library.
Assignment requirements
You are asked to write a Python program, with appropriate calls of scikit-learn functions, to fulfill the
following tasks. Although there is no strict rule on how to organize the code, each task should be
noted carefully.
1
Building the decision tree classifiers
This task conducts experiments on the designated train/test proportions listed above.
You need to fit an instance of sklearn.tree.DecisionTreeClassifier (with information
gain) to each training set and visualize the resulting decision tree using graphviz.
How do you interpret the classification report and the confusion matrix? From that, make your
own comments on the performances of those decision tree classifiers.
References
Grading
Notice