Package org.tribuo.regression.rtree

Class IndependentRegressionTreeModel

java.lang.Object

org.tribuo.Model<T>

org.tribuo.SparseModel<T>

org.tribuo.common.tree.TreeModel<Regressor>

org.tribuo.regression.rtree.IndependentRegressionTreeModel

All Implemented Interfaces:

com.oracle.labs.mlrg.olcut.provenance.Provenancable<org.tribuo.provenance.ModelProvenance>, Serializable

public final class IndependentRegressionTreeModel
extends TreeModel<Regressor>

A Model wrapped around a list of decision tree root Nodes used to generate independent predictions for each dimension in a regression.

See Also:

• Serialized Form

Field Summary

Fields inherited from class org.tribuo.Model

ALL_OUTPUTS, BIAS_FEATURE

Method Summary

Modifier and Type	Method	Description
int	getDepth()	Probes the trees to find the depth.
Optional<org.tribuo.Excuse<Regressor>>	getExcuse(org.tribuo.Example<Regressor> example)
Set<String>	getFeatures()
Node<Regressor>	getRoot()	Returns null, as this model contains multiple roots, one per regression output dimension.
Map<String,Node<Regressor>>	getRoots()	Returns an unmodifiable view on the root node collection.
Map<String,List<com.oracle.labs.mlrg.olcut.util.Pair<String,Double>>>	getTopFeatures(int n)
org.tribuo.Prediction<Regressor>	predict(org.tribuo.Example<Regressor> example)
String	toString()

Methods inherited from class org.tribuo.SparseModel

copy, getActiveFeatures

Methods inherited from class org.tribuo.Model

castModel, generatesProbabilities, getExcuses, getFeatureIDMap, getName, getOutputIDInfo, getProvenance, predict, predict, setName, validate

Methods inherited from class java.lang.Object

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

Method Details

getDepth

public int getDepth()

Probes the trees to find the depth.

Overrides:

getDepth in class TreeModel<Regressor>

Returns:

The maximum depth across the trees.

predict

public org.tribuo.Prediction<Regressor> predict(org.tribuo.Example<Regressor> example)

Overrides:

predict in class TreeModel<Regressor>

getTopFeatures

public Map<String,List<com.oracle.labs.mlrg.olcut.util.Pair<String,Double>>> getTopFeatures(int n)

Overrides:

getTopFeatures in class TreeModel<Regressor>

getExcuse

public Optional<org.tribuo.Excuse<Regressor>> getExcuse(org.tribuo.Example<Regressor> example)

Overrides:

getExcuse in class TreeModel<Regressor>

getFeatures

public Set<String> getFeatures()

Overrides:

getFeatures in class TreeModel<Regressor>

toString

public String toString()

Overrides:

toString in class TreeModel<Regressor>

getRoots

public Map<String,Node<Regressor>> getRoots()

Returns an unmodifiable view on the root node collection.

The nodes themselves are immutable.

Returns:

The root node collection.

getRoot

public Node<Regressor> getRoot()

Returns null, as this model contains multiple roots, one per regression output dimension.

Use getRoots() instead.

Overrides:

getRoot in class TreeModel<Regressor>

Returns:

null.