hex.genmodel.algos.tree

Class SharedTreeGraph

java.lang.Object

hex.genmodel.algos.tree.SharedTreeGraph

public class SharedTreeGraph
extends java.lang.Object

Graph for representing a GBM or DRF forest. A graph contains subgraphs (trees).

Field Summary

Fields

Modifier and Type	Field and Description
java.util.ArrayList<SharedTreeSubgraph>	subgraphArray

Constructor Summary

Constructors

Constructor and Description
SharedTreeGraph() Make a new forest.

Method Summary

Modifier and Type	Method and Description
boolean	equals(java.lang.Object o)
int	hashCode()
SharedTreeSubgraph	makeSubgraph(java.lang.String name) Make a new tree.
void	print() Debug printout of graph structure.
void	printDot(java.io.PrintStream os, int maxLevelsToPrintPerEdge, boolean detail, java.lang.String optionalTitle, PrintMojo.PrintTreeOptions treeOptions) Print graph output in a format readable by dot (graphviz).
java.util.List<java.util.Map<java.lang.String,java.lang.Object>>	toJson()
SharedTreeNode	walkNodes(int subgraphId, java.lang.String path)

Methods inherited from class java.lang.Object

clone, finalize, getClass, notify, notifyAll, toString, wait, wait, wait

Field Detail

subgraphArray

public final java.util.ArrayList<SharedTreeSubgraph> subgraphArray

Constructor Detail

SharedTreeGraph

public SharedTreeGraph()

Make a new forest.

Method Detail

makeSubgraph

public SharedTreeSubgraph makeSubgraph(java.lang.String name)

Make a new tree.

Parameters:

name - Tree name.

Returns:

The new tree.

print

public void print()

Debug printout of graph structure. For developer use only.

walkNodes

public SharedTreeNode walkNodes(int subgraphId,
                                java.lang.String path)

printDot

public void printDot(java.io.PrintStream os,
                     int maxLevelsToPrintPerEdge,
                     boolean detail,
                     java.lang.String optionalTitle,
                     PrintMojo.PrintTreeOptions treeOptions)

Print graph output in a format readable by dot (graphviz).

Parameters:

os - Stream to write the output to

maxLevelsToPrintPerEdge - Limit the number of individual categorical level names printed per edge

detail - include addtional node detail information

optionalTitle - Optional title to override the default

treeOptions - object of PrintTreeOptions to control how trees are printed in terms of font size and number of decimal places for numerical values

toJson

public java.util.List<java.util.Map<java.lang.String,java.lang.Object>> toJson()

equals

public boolean equals(java.lang.Object o)

Overrides:

equals in class java.lang.Object

hashCode

public int hashCode()

Overrides:

hashCode in class java.lang.Object