hex.genmodel.algos.tree

Interface SharedTreeGraphConverter

All Known Subinterfaces:

TreeBackedMojoModel

All Known Implementing Classes:

DrfMojoModel, GbmMojoModel, IsolationForestMojoModel, SharedTreeMojoModel, SharedTreeMojoModelWithContributions, UpliftDrfMojoModel

public interface SharedTreeGraphConverter

Implementors of this interface are able to convert internal tree representation to a shared representation.

Method Summary

Modifier and Type	Method and Description
SharedTreeGraph	convert(int treeNumber, java.lang.String treeClass)
SharedTreeGraph	convert(int treeNumber, java.lang.String treeClass, ConvertTreeOptions options) Converts internal tree representation to a shared representation.

Method Detail

convert

SharedTreeGraph convert(int treeNumber,
                        java.lang.String treeClass,
                        ConvertTreeOptions options)

Converts internal tree representation to a shared representation.

Parameters:

treeNumber - Number of the tree in the model to convert

treeClass - Tree's class. If not specified, all the classes form a forest in the resulting SharedTreeGraph

options - Allows to fine-tune the conversion process (eg. disable some internal consistency self-checks)

Returns:

An instance of SharedTreeGraph containing a single tree or a forest of multiple trees.

convert

SharedTreeGraph convert(int treeNumber,
                        java.lang.String treeClass)