Package org.apache.spark.ml.tree

Class DecisionTreeModelReadWrite.NodeData$

Object

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

All Implemented Interfaces:

Serializable

Enclosing class:

DecisionTreeModelReadWrite

public static class DecisionTreeModelReadWrite.NodeData$
extends Object
implements Serializable

See Also:

• Serialized Form

Field Summary

Fields

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

Constructor Summary

Constructors

Constructor	Description
NodeData$()

Method Summary

Modifier and Type	Method	Description
scala.Tuple2<scala.collection.immutable.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 java.lang.Object

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

Field Details

MODULE$

public static final DecisionTreeModelReadWrite.NodeData$ MODULE$

Static reference to the singleton instance of this Scala object.

Constructor Details

NodeData$

public NodeData$()

Method Details

build

public scala.Tuple2<scala.collection.immutable.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)