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, org.tribuo.protos.ProtoSerializable<org.tribuo.protos.core.ModelProto>

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

Modifier and Type	Field	Description
static final int	CURRENT_VERSION	Protobuf serialization version.

Fields inherited from class org.tribuo.Model

ALL_OUTPUTS, BIAS_FEATURE

Fields inherited from interface org.tribuo.protos.ProtoSerializable

DESERIALIZATION_METHOD_NAME, PROVENANCE_SERIALIZER

Method Summary

Modifier and Type	Method	Description
static IndependentRegressionTreeModel	deserializeFromProto(int version, String className, com.google.protobuf.Any message)	Deserialization factory.
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)
org.tribuo.protos.core.ModelProto	serialize()
String	toString()

Methods inherited from class org.tribuo.common.tree.TreeModel

countNodes

Methods inherited from class org.tribuo.SparseModel

copy, getActiveFeatures

Methods inherited from class org.tribuo.Model

castModel, deserialize, deserializeFromFile, deserializeFromStream, generatesProbabilities, getExcuses, getFeatureIDMap, getName, getOutputIDInfo, getProvenance, predict, predict, serializeToFile, serializeToStream, setName, validate

Methods inherited from class java.lang.Object

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

Field Details

CURRENT_VERSION

public static final int CURRENT_VERSION

Protobuf serialization version.

See Also:

• Constant Field Values

Method Details

deserializeFromProto

public static IndependentRegressionTreeModel deserializeFromProto(int version,
 String className,
 com.google.protobuf.Any message)
                                                           throws com.google.protobuf.InvalidProtocolBufferException

Deserialization factory.

Parameters:

version - The serialized object version.

className - The class name.

message - The serialized data.

Returns:

The deserialized object.

Throws:

com.google.protobuf.InvalidProtocolBufferException - If the protobuf could not be parsed from the message.

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.

serialize

public org.tribuo.protos.core.ModelProto serialize()

Specified by:

serialize in interface org.tribuo.protos.ProtoSerializable<org.tribuo.protos.core.ModelProto>

Overrides:

serialize in class TreeModel<Regressor>