Class LinearSGDModel
java.lang.Object
org.tribuo.Model<MultiLabel>
org.tribuo.common.sgd.AbstractSGDModel<MultiLabel>
org.tribuo.common.sgd.AbstractLinearSGDModel<MultiLabel>
org.tribuo.multilabel.sgd.linear.LinearSGDModel
- All Implemented Interfaces:
com.oracle.labs.mlrg.olcut.provenance.Provenancable<ModelProvenance>
,Serializable
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:
-
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 TypeMethodDescriptionprotected 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.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
-
Method Details
-
predict
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 classModel<MultiLabel>
- Parameters:
example
- the example to predict.- Returns:
- the result of the prediction.
-
getDimensionName
Description copied from class:AbstractLinearSGDModel
Gets the name of the indexed output dimension.- Specified by:
getDimensionName
in classAbstractLinearSGDModel<MultiLabel>
- Parameters:
index
- The output dimension index.- Returns:
- The name of the requested output dimension.
-
copy
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 classModel<MultiLabel>
- Parameters:
newName
- The new name.newProvenance
- The new provenance.- Returns:
- A copy of the model.
-