diff options
author | José Fonseca <jfonseca@vmware.com> | 2010-02-14 11:15:19 +0000 |
---|---|---|
committer | José Fonseca <jfonseca@vmware.com> | 2010-02-14 11:16:41 +0000 |
commit | 50812e633fa433937c7ba885fa334f44ec0bad58 (patch) | |
tree | fae5850bae8bdb6ab0ebd466f8f61b1a51d73d93 /progs/gallium/python/tests/tree.py | |
parent | 0a3e3621752be5b054739ce606cad4f77cb65049 (diff) |
progs/gallium/python: New home for python statetracker scripts.
Diffstat (limited to 'progs/gallium/python/tests/tree.py')
-rwxr-xr-x | progs/gallium/python/tests/tree.py | 23 |
1 files changed, 23 insertions, 0 deletions
diff --git a/progs/gallium/python/tests/tree.py b/progs/gallium/python/tests/tree.py new file mode 100755 index 0000000000..0c1bcda4cf --- /dev/null +++ b/progs/gallium/python/tests/tree.py @@ -0,0 +1,23 @@ +#!/usr/bin/env python +# +# See also: +# http://www.ailab.si/orange/doc/ofb/c_otherclass.htm + +import os.path +import sys + +import orange +import orngTree + +for arg in sys.argv[1:]: + name, ext = os.path.splitext(arg) + + data = orange.ExampleTable(arg) + + tree = orngTree.TreeLearner(data, sameMajorityPruning=1, mForPruning=2) + + orngTree.printTxt(tree) + + file(name+'.txt', 'wt').write(orngTree.dumpTree(tree) + '\n') + + orngTree.printDot(tree, fileName=name+'.dot', nodeShape='ellipse', leafShape='box') |