org.apache.spark.ml.tree

Class DecisionTreeModelReadWrite

Object

org.apache.spark.ml.tree.DecisionTreeModelReadWrite

public class DecisionTreeModelReadWrite
extends Object

Helper classes for tree model persistence

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	DecisionTreeModelReadWrite.NodeData Info for a Node
static class	DecisionTreeModelReadWrite.NodeData$
static class	DecisionTreeModelReadWrite.SplitData Info for a Split
static class	DecisionTreeModelReadWrite.SplitData$

Constructor Summary

Constructors

Constructor and Description
DecisionTreeModelReadWrite()

Method Summary

Modifier and Type	Method and Description
static Node	buildTreeFromNodes(DecisionTreeModelReadWrite.NodeData[] data, String impurityType) Given all data for all nodes in a tree, rebuild the tree.
static Node	loadTreeNodes(String path, org.apache.spark.ml.util.DefaultParamsReader.Metadata metadata, SparkSession sparkSession) Load a decision tree from a file.

Methods inherited from class Object

equals, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

DecisionTreeModelReadWrite

public DecisionTreeModelReadWrite()

Method Detail

loadTreeNodes

public static Node loadTreeNodes(String path,
                                 org.apache.spark.ml.util.DefaultParamsReader.Metadata metadata,
                                 SparkSession sparkSession)

Load a decision tree from a file.

Parameters:

path - (undocumented)

metadata - (undocumented)

sparkSession - (undocumented)

Returns:

Root node of reconstructed tree

buildTreeFromNodes

public static Node buildTreeFromNodes(DecisionTreeModelReadWrite.NodeData[] data,
                                      String impurityType)

Given all data for all nodes in a tree, rebuild the tree.

Parameters:

data - Unsorted node data

impurityType - Impurity type for this tree

Returns:

Root node of reconstructed tree