org.apache.spark.ml.regression

Interface FactorizationMachines

All Superinterfaces:

FactorizationMachinesParams, HasFeaturesCol, HasFitIntercept, HasLabelCol, HasMaxIter, HasPredictionCol, HasRegParam, HasSeed, HasSolver, HasStepSize, HasTol, HasWeightCol, Identifiable, Params, PredictorParams, java.io.Serializable

All Known Implementing Classes:

FMClassifier, FMRegressor

public interface FactorizationMachines
extends FactorizationMachinesParams

Method Summary

Modifier and Type	Method and Description
Vector	initCoefficients(int numFeatures)
scala.Tuple2<Vector,double[]>	trainImpl(RDD<scala.Tuple2<Object,Vector>> data, int numFeatures, String loss)

Methods inherited from interface org.apache.spark.ml.regression.FactorizationMachinesParams

factorSize, fitLinear, getFactorSize, getFitLinear, getInitStd, getMiniBatchFraction, initStd, miniBatchFraction, solver

Methods inherited from interface org.apache.spark.ml.PredictorParams

validateAndTransformSchema

Methods inherited from interface org.apache.spark.ml.param.shared.HasLabelCol

getLabelCol, labelCol

Methods inherited from interface org.apache.spark.ml.param.shared.HasFeaturesCol

featuresCol, getFeaturesCol

Methods inherited from interface org.apache.spark.ml.param.shared.HasPredictionCol

getPredictionCol, predictionCol

Methods inherited from interface org.apache.spark.ml.param.Params

clear, copy, copyValues, defaultCopy, defaultParamMap, explainParam, explainParams, extractParamMap, extractParamMap, get, getDefault, getOrDefault, getParam, hasDefault, hasParam, isDefined, isSet, onParamChange, paramMap, params, set, set, set, setDefault, setDefault, shouldOwn

Methods inherited from interface org.apache.spark.ml.util.Identifiable

toString, uid

Methods inherited from interface org.apache.spark.ml.param.shared.HasMaxIter

getMaxIter, maxIter

Methods inherited from interface org.apache.spark.ml.param.shared.HasStepSize

getStepSize, stepSize

Methods inherited from interface org.apache.spark.ml.param.shared.HasTol

getTol, tol

Methods inherited from interface org.apache.spark.ml.param.shared.HasSolver

getSolver

Methods inherited from interface org.apache.spark.ml.param.shared.HasSeed

getSeed, seed

Methods inherited from interface org.apache.spark.ml.param.shared.HasFitIntercept

fitIntercept, getFitIntercept

Methods inherited from interface org.apache.spark.ml.param.shared.HasRegParam

getRegParam, regParam

Methods inherited from interface org.apache.spark.ml.param.shared.HasWeightCol

getWeightCol, weightCol

Method Detail

initCoefficients

Vector initCoefficients(int numFeatures)

trainImpl

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