Package org.apache.spark.ml.regression

Class FMRegressionModel

Object
org.apache.spark.ml.PipelineStage
org.apache.spark.ml.Transformer
org.apache.spark.ml.Model<M>
org.apache.spark.ml.PredictionModel<FeaturesType,M>
org.apache.spark.ml.regression.RegressionModel<Vector,FMRegressionModel>
org.apache.spark.ml.regression.FMRegressionModel
All Implemented Interfaces:
Serializable, org.apache.spark.internal.Logging, Params, HasFeaturesCol, HasFitIntercept, HasLabelCol, HasMaxIter, HasPredictionCol, HasRegParam, HasSeed, HasSolver, HasStepSize, HasTol, HasWeightCol, PredictorParams, FactorizationMachinesParams, FMRegressorParams, Identifiable, MLWritable
public class FMRegressionModel extends RegressionModel<Vector,FMRegressionModel> implements FMRegressorParams, MLWritable
Model produced by FMRegressor.
See Also:

  • Nested Class Summary

    Nested classes/interfaces inherited from interface org.apache.spark.internal.Logging

    org.apache.spark.internal.Logging.LogStringContext, org.apache.spark.internal.Logging.SparkShellLoggingFilter

  • Method Summary

    Modifier and Type
    Method
    Description
    FMRegressionModel
    copy(ParamMap extra)
    Creates a copy of this instance with the same UID and some extra params.
    Matrix
    factors()
     
    final IntParam
    factorSize()
    Param for dimensionality of the factors (&gt;= 0)
    final BooleanParam
    fitIntercept()
    Param for whether to fit an intercept term.
    final BooleanParam
    fitLinear()
    Param for whether to fit linear term (aka 1-way term)
    final DoubleParam