org.apache.ignite.ml.tree.randomforest.data

Class TreeRoot

java.lang.Object

org.apache.ignite.ml.tree.randomforest.data.TreeRoot

All Implemented Interfaces:

Serializable, Function<Vector,Double>, IgniteFunction<Vector,Double>, Model<Vector,Double>

public class TreeRoot
extends Object
implements Model<Vector,Double>

Tree root class.

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
TreeRoot(TreeNode root, Set<Integer> usedFeatures) Create an instance of TreeRoot.

Method Summary

Modifier and Type	Method and Description
Double	apply(Vector vector)
List<TreeNode>	getLeafs()
TreeNode	getRootNode()
Set<Integer>	getUsedFeatures()

Methods inherited from class java.lang.Object

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

Methods inherited from interface org.apache.ignite.ml.Model

combine, toString

Methods inherited from interface java.util.function.Function

andThen, compose, identity

Constructor Detail

TreeRoot

public TreeRoot(TreeNode root,
                Set<Integer> usedFeatures)

Create an instance of TreeRoot.

Parameters:

root - Root.

usedFeatures - Used features.

Method Detail

apply

public Double apply(Vector vector)

Specified by:

apply in interface Function<Vector,Double>

getUsedFeatures

public Set<Integer> getUsedFeatures()

getRootNode

public TreeNode getRootNode()

getLeafs

public List<TreeNode> getLeafs()

Returns:

all leafs in tree.