rdkit.ML.DecTree.TreeVis module

functionality for drawing trees on sping canvases

rdkit.ML.DecTree.TreeVis.CalcTreeNodeSizes(node)

Recursively calculate the total number of nodes under us.

results are set in node.totNChildren for this node and everything underneath it.

rdkit.ML.DecTree.TreeVis.CalcTreeWidth(tree)

rdkit.ML.DecTree.TreeVis.DrawTree(tree, canvas, nRes=2, scaleLeaves=False, allowShrink=True, showPurity=False)

rdkit.ML.DecTree.TreeVis.DrawTreeNode(node, loc, canvas, nRes=2, scaleLeaves=False, showPurity=False)

Recursively displays the given tree node and all its children on the canvas

rdkit.ML.DecTree.TreeVis.ResetTree(tree)

rdkit.ML.DecTree.TreeVis.SetNodeScales(node)

class rdkit.ML.DecTree.TreeVis.VisOpts

Bases: object

circColor = Color(0.60,0.60,0.90)

circRad = 10

highlightColor = Color(1.00,1.00,0.40)

highlightWidth = 2

horizOffset = 10

labelFont = Font(10,0,0,0,'helvetica')

lineColor = Color(0.00,0.00,0.00)

lineWidth = 2

maxCircRad = 16

minCircRad = 4

outlineColor = Color(-1.00,-1.00,-1.00)

terminalEmptyColor = Color(0.80,0.80,0.20)

terminalOffColor = Color(0.20,0.20,0.20)

terminalOnColor = Color(0.80,0.80,0.80)

vertOffset = 50