SkillAgentSearch skills...

DecisionTrees

A python implementation of the CART algorithm for decision trees

Install / Use

/learn @lucksd356/DecisionTrees
About this skill

Quality Score

0/100

Supported Platforms

Universal

README

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
  • pydotplus: python interface to GraphVis dot language, http://pydotplus.readthedocs.io/index.html
  • GraphViz: to render the graphs, http://www.graphviz.org/
GraphViz plot of the classification tree for the Fisher irises data

gini tree plot

Related Skills

View on GitHub
GitHub Stars47
CategoryDevelopment
Updated1y ago
Forks36

Languages

Python

Security Score

80/100

Audited on Nov 11, 2024

No findings