org.tribuo.multilabel.sgd.linear

Class LinearSGDModel

java.lang.Object

org.tribuo.Model<T>

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

org.tribuo.common.sgd.AbstractLinearSGDModel<MultiLabel>

org.tribuo.multilabel.sgd.linear.LinearSGDModel

All Implemented Interfaces:

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

public class LinearSGDModel
extends AbstractLinearSGDModel<MultiLabel>

The inference time version of a multi-label linear model trained using SGD.

See:

 Bottou L.
 "Large-Scale Machine Learning with Stochastic Gradient Descent"
 Proceedings of COMPSTAT, 2010.
 

See Also:

Serialized Form

Nested Class Summary

Nested classes/interfaces inherited from class org.tribuo.common.sgd.AbstractSGDModel

AbstractSGDModel.PredAndActive

Field Summary

Fields inherited from class org.tribuo.common.sgd.AbstractSGDModel

addBias, modelParameters

Fields inherited from class org.tribuo.Model

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

Method Summary

Modifier and Type	Method and Description
protected LinearSGDModel	copy(String newName, ModelProvenance newProvenance) Copies a model, replacing it's provenance and name with the supplied values.
protected String	getDimensionName(int index) Gets the name of the indexed output dimension.
Prediction<MultiLabel>	predict(Example<MultiLabel> example) Uses the model to predict the output for a single example.

Methods inherited from class org.tribuo.common.sgd.AbstractLinearSGDModel

getExcuse, getTopFeatures, getWeightsCopy

Methods inherited from class org.tribuo.common.sgd.AbstractSGDModel

getModelParameters, predictSingle

Methods inherited from class org.tribuo.Model

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

Methods inherited from class java.lang.Object

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

Method Detail

predict

public Prediction<MultiLabel> predict(Example<MultiLabel> example)

Description copied from class: Model

Uses the model to predict the output for a single example.

predict does not mutate the example.

Throws IllegalArgumentException if the example has no features or no feature overlap with the model.

Specified by:

predict in class Model<MultiLabel>

Parameters:

example - the example to predict.

Returns:

the result of the prediction.

getDimensionName

protected String getDimensionName(int index)

Description copied from class: AbstractLinearSGDModel

Gets the name of the indexed output dimension.

Specified by:

getDimensionName in class AbstractLinearSGDModel<MultiLabel>

Parameters:

index - The output dimension index.

Returns:

The name of the requested output dimension.

copy

protected LinearSGDModel copy(String newName,
                              ModelProvenance newProvenance)

Description copied from class: Model

Copies a model, replacing it's provenance and name with the supplied values.

Used to provide the provenance removal functionality.

Specified by:

copy in class Model<MultiLabel>

Parameters:

newName - The new name.

newProvenance - The new provenance.

Returns:

A copy of the model.