Remember from lecture, we learned that we can use decision tree to solve classification and regression problem. Mostly we focus on classification. In problem 1 we used KNN to do classification. We could also use decision tree algorithm to do the same job. For Decision Tree, you will be asked to implement ID3 algorithm. It’s guaranteed that all features are discrete in this assignment. After finishing the implementation, you can use dt_test_script.py to test the correctness of your functions.
2.1.1 Information Gain
- In ID3 algorithm, we use Entropy to measure the uncertainty in the data set. We use Information Gain to measure the quality of a split.
- Entropy:
- Information_Gain:
- see more detail on ID3 Algorithm In this section, you need to implement Information_Gain function on utils.py.
2.1.2 Grow decision Tree and predict (25 points)
- In ID3 algorithm, we use the largest information gain to split the set S. Please consult the Lecture 2 notes page 23.
- Implement TreeNode split function and TreeNode predict function in hw1_dt.py:
- TreeNode.split
- In TreeNode class, variable features represents all data points in current TreeNode. Variable labels represents corresponding labels for all data points. Variable children is a list of TreeNode after split the current node on best attribute. This should be a recursive process that once we call the split function, the TreeNode will keep spliting until we get the whole decision tree.
- When there is a tie of information gain on comparing all attributes of current TreeNode, always choose the attribute which has more attribute values. If they have same number of attribute values, use the one with small index.
- Build child list of each TreeNode with increasing order on attribute values, the order matters because it will be used when comparing two decision trees.
- TreeNode.predict
- This function will be called once you have got the decision tree. It will take one single data point as a parameter, your code should process that data point and go through your tree to a leaf node and make prediction. This function should return a predicted lable.
- TreeNode.split
- You don’t need to implement Decision Tree predict and train function in hw1_dt.py. We will provide these two function in the statercode. Reading the train and predict function would help you understand functions that you need to implement.
- DecisionTree.train
- DecisionTree.predict
- This is the comarision logic we will use to check your decision tree.
Part 2.2 Pruning Decision Tree
Sometimes, in order to prevent overfitting. We need to prune our Decision Tree. There are several approaches to avoid overfitting in building decision trees.
- Pre-pruning: stop growing the tree earlier, before it perfectly classifies the training set.
- Post-pruning: grows decision tree completely on training set, and then post prune decision tree.
Practically, the second approach post-pruning is more useful, also it is not easy to precisely estimate when to stop growing the tree by using pre-pruning. We will use Reduced Error Pruning, as one of Post-pruning in this assignment.
For Pruning of Decision Tree, you can refer Reduce Error Pruning and P69 of Textbook: Machine Learning -Tom Mitchell.
2.2.1 Reduce Error Pruning
Hint: in this part, you can add other parameters or functions in TreeNode class and DecisionTree class to help you in pruning the decision tree. But your changes should not influence results of previous parts. Implement the reduced_error_pruning function in util.py.
Good Luck! 🙂
Rate this product
# calculate information_gain according to branches seperated by one feature
# input:
-branches: List[List[int]] num_attribute_values*num_classes : Count of data points belonging to each attribute and class.
def Information_Gain(S, branches):
-S: float Entropy of current state
e.g. [[1,2],[1,1][2,0]] represents there are three attribute values. For attribute 0, there is one data point that belongs to class 0 and two data points that belong to class 1.
# return: float
if dTree_node.splittable != my_dTree_node.splittable:
if dTree_node.cls_max != my_dTree_node.cls_max:
if dTree_node.splittable:
return False
for idx, child in enumerate(dTree_node.children):
if not compareDecisionTree(child, my_dTree_node.children[idx]):
return True
def compareDecisionTree(dTree_node, my_dTree_node):
return False
return False
if len(dTree_node.children) != len(my_dTree_node.children):
return False
0. Split data into training and validation sets.
1. Grow the decision tree until further pruning is harmful:
2. Evaluate decision tree on validation dataset.
Reduced Error Pruning
3. Greedily remove TreeNode (from leaf to root) that most improves prediction accuracy on validation set.
# input:
def reduced_error_pruning(decitionTree):
- decitionTree: decitionTree trained based on training data set.
- X_test: List[List[any]] test data, num_cases*num_attributes
- y_test: List[any] test labels, num_cases*1
Reviews
There are no reviews yet.