Changeset 7890:010f0243a070 in orange


Ignore:
Timestamp:
05/04/11 11:25:48 (3 years ago)
Author:
markotoplak
Branch:
default
Convert:
ebd34e56866a04b32df9da25bd69bfa118ffdd7f
Message:

_TreeDumper can work with (Tree)Nodes directly. Fixes #719.

File:
1 edited

Legend:

Unmodified
Added
Removed
  • orange/Orange/classification/tree.py

    r7808 r7890  
    27472747         (re_I, replaceI) ] 
    27482748 
     2749    def node(self): 
     2750        return self.tree.tree if "tree" in self.tree.__dict__ else self.tree 
     2751 
    27492752    def __init__(self, leafStr, nodeStr, stringFormats, minExamples,  
    27502753        maxDepth, simpleFirst, tree, **kw): 
     
    27592762            self.leafStr = leafStr 
    27602763        else: 
    2761             if tree.classVar.varType == Orange.data.Type.Discrete: 
     2764            if self.node().node_classifier.classVar.varType == \ 
     2765                    Orange.data.Type.Discrete: 
    27622766                self.leafStr = "%V (%^.2m%)" 
    27632767            else: 
     
    28442848 
    28452849    def dumpTree(self): 
     2850        node = self.node() 
    28462851        if self.nodeStr: 
    28472852            lev, res = 1, "root: %s\n" % \ 
    2848                 self.formatString(self.nodeStr, self.tree.tree, None) 
     2853                self.formatString(self.nodeStr, node, None) 
    28492854            self.maxDepth += 1 
    28502855        else: 
    28512856            lev, res = 0, "" 
    2852         return res + self.dumpTree0(self.tree.tree, None, lev) 
     2857        return res + self.dumpTree0(node, None, lev) 
    28532858         
    28542859 
     
    28842889    def dotTree(self, internalName="n"): 
    28852890        self.fle.write("digraph G {\n") 
    2886         self.dotTree0(self.tree.tree, None, internalName) 
     2891        self.dotTree0(self.node(), None, internalName) 
    28872892        self.fle.write("}\n") 
    28882893 
Note: See TracChangeset for help on using the changeset viewer.