sim-pez / decision_tree

Decision tree ML algorithm

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Description

This is an implementation of decision tree algorithm described in the textbook "Artificial Intelligence: A Modern Approach" written by Stuart J. Russell and Peter Norvig.

The impurity measure is entropy. There is also a pruning technique on DNF of every possible path based on errors on validation set.

Hope this project would be useful for someone :)

Dataset

You can download Adult dataset here

Acknowledgments

Intelligenza Artificiale - Computer Engineering Bachelor's Degree @University of Florence

About

Decision tree ML algorithm


Languages

Language:Jupyter Notebook 100.0%