Skip to content

RiccardoDiGuida/DecisionTree

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

17 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

DecisionTree

Some days ago I watched this lecture by Nando De Freitas on Decision Trees so I decided to try my personal implementation of the algorithm (also to learn how to use cmake). It supports categorical and continuous variables as descriptors. The labels are not constrained on two levels only. It does not rely on any external library.

The criterion to select the best variable to split the dataset is based on the achievement of the maximum information gain calculated on entropy.

In order to build you can simply access the project root directory and type

mkdir build 
cd build 
cmake ..
make 

For the moment it is only possible to build the trees. I will provide error estimates in the near future.

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published