/DecisionTree

Implements an ID3-like decision-tree learner for classification

Primary LanguagePython

DecisionTree

Implements an ID3-like decision-tree learner for classification

Requirements

Python 2.7

Important Scripts

  • dt-learn: Script that calls dt-learn.py
  • dt-learn.py: Contains the implementation of the decision-tree
  • *.arff: Sample training and testing files

Usage

Command to run the classifier: ./dt-learn train-set-file test-set-file m

m specifies the minimum number of training instances that need to be present to continue building the tree. If the number is less than 'm', then the node is converted into a leaf. This is one of the stopping criteria implemented apart from the normal ID3 based stopping criteria