org.apache.spark.ml.tree

Class DecisionTreeModelReadWrite.NodeData$

Object

org.apache.spark.ml.tree.DecisionTreeModelReadWrite.NodeData$

All Implemented Interfaces:

java.io.Serializable

Enclosing class:

DecisionTreeModelReadWrite

public static class DecisionTreeModelReadWrite.NodeData$
extends Object
implements scala.Serializable

See Also:

Serialized Form

Field Summary

Fields

Modifier and Type	Field and Description
static DecisionTreeModelReadWrite.NodeData$	MODULE$ Static reference to the singleton instance of this Scala object.

Constructor Summary

Constructors

Constructor and Description
NodeData$()

Method Summary

Modifier and Type	Method and Description
scala.Tuple2<scala.collection.Seq<DecisionTreeModelReadWrite.NodeData>,Object>	build(Node node, int id) Create DecisionTreeModelReadWrite.NodeData instances for this node and all children.
int	inferNumPartitions(long numNodes) When save a tree model, infer the number of partitions based on number of nodes.

Methods inherited from class Object

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

Field Detail

MODULE$

public static final DecisionTreeModelReadWrite.NodeData$ MODULE$

Static reference to the singleton instance of this Scala object.

Constructor Detail

NodeData$

public NodeData$()

Method Detail

build

public scala.Tuple2<scala.collection.Seq<DecisionTreeModelReadWrite.NodeData>,Object> build(Node node,
                                                                                            int id)

Create DecisionTreeModelReadWrite.NodeData instances for this node and all children.

Parameters:

id - Current ID. IDs are assigned via a pre-order traversal.

node - (undocumented)

Returns:

(sequence of nodes in pre-order traversal order, largest ID in subtree) The nodes are returned in pre-order traversal (root first) so that it is easy to get the ID of the subtree's root node.

inferNumPartitions

public int inferNumPartitions(long numNodes)

When save a tree model, infer the number of partitions based on number of nodes.

Parameters:

numNodes - (undocumented)

Returns:

(undocumented)