src/gallium/tests/python/tests/tree.py

   1 #!/usr/bin/env python
   2 #
   3 # See also:
   4 #  http://www.ailab.si/orange/doc/ofb/c_otherclass.htm
   5
   6 import os.path
   7 import sys
   8
   9 import orange
  10 import orngTree
  11
  12 for arg in sys.argv[1:]:
  13     name, ext = os.path.splitext(arg)
  14
  15     data = orange.ExampleTable(arg)
  16
  17     tree = orngTree.TreeLearner(data, sameMajorityPruning=1, mForPruning=2)
  18
  19     orngTree.printTxt(tree)
  20
  21     file(name+'.txt', 'wt').write(orngTree.dumpTree(tree) + '\n')
  22
  23     orngTree.printDot(tree, fileName=name+'.dot', nodeShape='ellipse', leafShape='box')