public interface FactorizationMachines extends FactorizationMachinesParams
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) |
factorSize, fitLinear, getFactorSize, getFitLinear, getInitStd, getMiniBatchFraction, initStd, miniBatchFraction, solver
validateAndTransformSchema
getLabelCol, labelCol
featuresCol, getFeaturesCol
getPredictionCol, predictionCol
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
toString, uid
getMaxIter, maxIter
getStepSize, stepSize
fitIntercept, getFitIntercept
getRegParam, regParam
getWeightCol, weightCol