diff options
author | José Fonseca <[email protected]> | 2009-03-31 18:38:48 +0100 |
---|---|---|
committer | José Fonseca <[email protected]> | 2009-03-31 18:38:48 +0100 |
commit | f784906eb96ccda2062b53867a5d4bc52653396d (patch) | |
tree | 3e37ceac75f7b8fe758f91324de2419c2f422bfe /src | |
parent | fee78c0c1fcfc308c84ab8da1efcc98ed8afc889 (diff) |
python/test: Dump classification tree to text file too.
Diffstat (limited to 'src')
-rwxr-xr-x | src/gallium/state_trackers/python/tests/base.py | 2 | ||||
-rwxr-xr-x | src/gallium/state_trackers/python/tests/tree.py | 2 |
2 files changed, 4 insertions, 0 deletions
diff --git a/src/gallium/state_trackers/python/tests/base.py b/src/gallium/state_trackers/python/tests/base.py index 6b2e702955f..1fa7fe6f3b3 100755 --- a/src/gallium/state_trackers/python/tests/base.py +++ b/src/gallium/state_trackers/python/tests/base.py @@ -331,4 +331,6 @@ class TestResult: orngTree.printTxt(tree, maxDepth=4) + file(name+'.txt', 'wt').write(orngTree.dumpTree(tree)) + orngTree.printDot(tree, fileName=name+'.dot', nodeShape='ellipse', leafShape='box') diff --git a/src/gallium/state_trackers/python/tests/tree.py b/src/gallium/state_trackers/python/tests/tree.py index 14b3599c6d7..0c1bcda4cf2 100755 --- a/src/gallium/state_trackers/python/tests/tree.py +++ b/src/gallium/state_trackers/python/tests/tree.py @@ -18,4 +18,6 @@ for arg in sys.argv[1:]: orngTree.printTxt(tree) + file(name+'.txt', 'wt').write(orngTree.dumpTree(tree) + '\n') + orngTree.printDot(tree, fileName=name+'.dot', nodeShape='ellipse', leafShape='box') |