Interface FactorizationMachines

All Superinterfaces:
FactorizationMachinesParams, HasFeaturesCol, HasFitIntercept, HasLabelCol, HasMaxIter, HasPredictionCol, HasRegParam, HasSeed, HasSolver, HasStepSize, HasTol, HasWeightCol, Identifiable, Params, PredictorParams, Serializable, scala.Serializable
All Known Implementing Classes:
FMClassifier, FMRegressor

public interface FactorizationMachines extends FactorizationMachinesParams
  • Method Details

    • initCoefficients

      Vector initCoefficients(int numFeatures)
    • trainImpl

      scala.Tuple2<Vector,double[]> trainImpl(RDD<scala.Tuple2<Object,Vector>> data, int numFeatures, String loss)