bachtn / decision_tree

- implentation of the decision tree algorithm, different split critereon, some prune algorithm

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

decision_tree

  • Implementation of decision tree algorithm
  • Multiple Split metrics (entropy (Information gain, Gain Ratio), gini impurity)
  • Pruning algorithms (Naive pruning, Reduced Error Pruning (REP))
  • Benchmark of multiple algorithms

About

- implentation of the decision tree algorithm, different split critereon, some prune algorithm

License:GNU General Public License v3.0


Languages

Language:Jupyter Notebook 83.2%Language:Python 16.7%Language:Makefile 0.1%