Uses of Class
org.tribuo.common.sgd.FMParameters
Packages that use FMParameters
Package
Description
Provides an implementation of a classification factorization machine using Stochastic Gradient Descent.
Provides the base classes for models trained with stochastic gradient descent.
Provides an implementation of a multi-label classification factorization machine model using Stochastic Gradient Descent.
Provides an implementation of factorization machines for regression using Stochastic Gradient Descent.
-
Uses of FMParameters in org.tribuo.classification.sgd.fm
Methods in org.tribuo.classification.sgd.fm with parameters of type FMParametersModifier and TypeMethodDescriptionprotected FMClassificationModelFMClassificationTrainer.createModel(String name, ModelProvenance provenance, ImmutableFeatureMap featureMap, ImmutableOutputInfo<Label> outputInfo, FMParameters parameters) -
Uses of FMParameters in org.tribuo.common.sgd
Subclasses with type arguments of type FMParameters in org.tribuo.common.sgdModifier and TypeClassDescriptionclassAbstractFMTrainer<T extends Output<T>, U, V extends AbstractFMModel<T>>A trainer for a quadratic factorization machine model which uses SGD.Methods in org.tribuo.common.sgd that return FMParametersModifier and TypeMethodDescriptionFMParameters.copy()protected FMParametersAbstractFMTrainer.createParameters(int numFeatures, int numOutputs, SplittableRandom localRNG) Constructs the trainable parameters object, in this case aFMParameterscontaining a weight matrix for the feature weights and a series of weight matrices for the factorized feature representation.static FMParametersFMParameters.deserializeFromProto(int version, String className, com.google.protobuf.Any message) Deserialization factory.Constructors in org.tribuo.common.sgd with parameters of type FMParametersModifierConstructorDescriptionprotectedAbstractFMModel(String name, ModelProvenance provenance, ImmutableFeatureMap featureIDMap, ImmutableOutputInfo<T> outputIDInfo, FMParameters parameters, boolean generatesProbabilities) Constructs a factorization machine model trained via SGD. -
Uses of FMParameters in org.tribuo.multilabel.sgd.fm
Methods in org.tribuo.multilabel.sgd.fm with parameters of type FMParametersModifier and TypeMethodDescriptionprotected FMMultiLabelModelFMMultiLabelTrainer.createModel(String name, ModelProvenance provenance, ImmutableFeatureMap featureMap, ImmutableOutputInfo<MultiLabel> outputInfo, FMParameters parameters) -
Uses of FMParameters in org.tribuo.regression.sgd.fm
Methods in org.tribuo.regression.sgd.fm with parameters of type FMParametersModifier and TypeMethodDescriptionprotected FMRegressionModelFMRegressionTrainer.createModel(String name, ModelProvenance provenance, ImmutableFeatureMap featureMap, ImmutableOutputInfo<Regressor> outputInfo, FMParameters parameters)