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

Class TreeRoot

java.lang.Object

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

All Implemented Interfaces:

Serializable, AutoCloseable, IgniteModel<Vector,Double>, Model<Vector,Double>

public class TreeRoot
extends Object
implements IgniteModel<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
List<TreeNode>	getLeafs()
TreeNode	getRootNode()
Set<Integer>	getUsedFeatures()
Double	predict(Vector vector) Make a prediction for the specified input arguments.

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.IgniteModel

andBefore, andThen, andThen, close, combine, toString

Constructor Detail

TreeRoot

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

Create an instance of TreeRoot.

Parameters:

root - Root.

usedFeatures - Used features.

Method Detail

predict

public Double predict(Vector vector)

Make a prediction for the specified input arguments.

Specified by:

predict in interface Model<Vector,Double>

Parameters:

vector - Input arguments.

Returns:

Prediction result.

getUsedFeatures

public Set<Integer> getUsedFeatures()

getRootNode

public TreeNode getRootNode()

getLeafs

public List<TreeNode> getLeafs()

Returns:

All leafs in tree.