in sagemaker-spark-sdk/src/main/scala/com/amazonaws/services/sagemaker/sparksdk/algorithms/FactorizationMachinesSageMakerEstimator.scala [722:778]
def setFeatureDim(value: Int): this.type = set(featureDim, value)
def setMiniBatchSize(value: Int): this.type = set(miniBatchSize, value)
def setNumFactors(value: Int): this.type = set(numFactors, value)
def setEpochs(value: Int): this.type = set(epochs, value)
def setClipGradient(value: Double): this.type = set(clipGradient, value)
def setEps(value: Double): this.type = set(eps, value)
def setRescaleGrad(value: Double): this.type = set(rescaleGrad, value)
def setBiasLr(value: Double): this.type = set(biasLr, value)
def setLinearLr(value: Double): this.type = set(linearLr, value)
def setFactorsLr(value: Double): this.type = set(factorsLr, value)
def setBiasWd(value: Double): this.type = set(biasWd, value)
def setLinearWd(value: Double): this.type = set(linearWd, value)
def setFactorsWd(value: Double): this.type = set(factorsWd, value)
def setBiasInitMethod(value: String): this.type = set(biasInitMethod, value)
def setBiasInitScale(value: Double): this.type = set(biasInitScale, value)
def setBiasInitSigma(value: Double): this.type = set(biasInitSigma, value)
def setBiasInitValue(value: Double): this.type = set(biasInitValue, value)
def setLinearInitMethod(value: String): this.type = set(linearInitMethod, value)
def setLinearInitScale(value: Double): this.type = set(linearInitScale, value)
def setLinearInitSigma(value: Double): this.type = set(linearInitSigma, value)
def setLinearInitValue(value: Double): this.type = set(linearInitValue, value)
def setFactorsInitMethod(value: String): this.type = set(factorsInitMethod, value)
def setFactorsInitScale(value: Double): this.type = set(factorsInitScale, value)
def setFactorsInitSigma(value: Double): this.type = set(factorsInitSigma, value)
def setFactorsInitValue(value: Double): this.type = set(factorsInitValue, value)
// Check whether required hyper-parameters are set
override def transformSchema(schema: StructType): StructType = {
$(featureDim)
$(predictorType)
$(numFactors)
super.transformSchema(schema)
}