Skip to content

A python implementation of the CART algorithm for decision trees

License

Notifications You must be signed in to change notification settings

lucksd356/DecisionTrees

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

7 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

CART For Decision Trees

This is a python implementation of the CART algorithm for decision trees based on Michael Dorner's code, https://github.com/michaeldorner/DecisionTrees.

Summary of code changes
  1. Fixed a bug on lines 96 & 97 of the original code
  2. Added the option to read feature names from a header line
  3. Use the pydotplus package to generate a GraphViz dot script for the decision tree
Requirements
GraphViz plot of the classification tree for the Fisher irises data

gini tree plot

About

A python implementation of the CART algorithm for decision trees

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages