org.tribuo.common.sgd

Class AbstractSGDModel<T extends Output<T>>

java.lang.Object

org.tribuo.Model<T>

org.tribuo.common.sgd.AbstractSGDModel<T>

All Implemented Interfaces:

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

Direct Known Subclasses:

AbstractLinearSGDModel

public abstract class AbstractSGDModel<T extends Output<T>>
extends Model<T>

See Also:

Serialized Form

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
protected static class	AbstractSGDModel.PredAndActive A nominal tuple used to capture the prediction and the number of active features used by the model.

Field Summary

Fields

Modifier and Type	Field and Description
protected boolean	addBias
protected FeedForwardParameters	modelParameters The weights for this model.

Fields inherited from class org.tribuo.Model

ALL_OUTPUTS, BIAS_FEATURE, featureIDMap, generatesProbabilities, name, outputIDInfo, provenance, provenanceOutput

Constructor Summary

Constructors

Modifier	Constructor and Description
protected	AbstractSGDModel(String name, ModelProvenance provenance, ImmutableFeatureMap featureIDMap, ImmutableOutputInfo<T> outputIDInfo, FeedForwardParameters weights, boolean generatesProbabilities, boolean addBias) Constructs a linear model trained via SGD.

Method Summary

Modifier and Type	Method and Description
FeedForwardParameters	getModelParameters() Returns a copy of the model parameters.
protected AbstractSGDModel.PredAndActive	predictSingle(Example<T> example) Generates the dense vector prediction from the supplied example.

Methods inherited from class org.tribuo.Model

copy, copy, generatesProbabilities, getExcuse, getExcuses, getFeatureIDMap, getName, getOutputIDInfo, getProvenance, getTopFeatures, innerPredict, predict, predict, predict, setName, toString, validate

Methods inherited from class java.lang.Object

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

Field Detail

modelParameters

protected FeedForwardParameters modelParameters

The weights for this model.

addBias

protected boolean addBias

Constructor Detail

AbstractSGDModel

protected AbstractSGDModel(String name,
                           ModelProvenance provenance,
                           ImmutableFeatureMap featureIDMap,
                           ImmutableOutputInfo<T> outputIDInfo,
                           FeedForwardParameters weights,
                           boolean generatesProbabilities,
                           boolean addBias)

Constructs a linear model trained via SGD.

Parameters:

name - The model name.

provenance - The model provenance.

featureIDMap - The feature domain.

outputIDInfo - The output domain.

weights - The model weights.

generatesProbabilities - Does this model generate probabilities?

addBias - Should the model add a bias feature to the feature vector?

Method Detail

predictSingle

protected AbstractSGDModel.PredAndActive predictSingle(Example<T> example)

Generates the dense vector prediction from the supplied example.

Parameters:

example - The example to use for prediction.

Returns:

The prediction and the number of features involved.

getModelParameters

public FeedForwardParameters getModelParameters()

Returns a copy of the model parameters.

Returns:

A copy of the model parameters.