def setFeatureDim()

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)
  }