def setK()

in sagemaker-spark-sdk/src/main/scala/com/amazonaws/services/sagemaker/sparksdk/algorithms/KMeansSageMakerEstimator.scala [302:336]

• 20 lines of code
• 1 McCabe index (conditional complexity)

  def setK(value : Int) : this.type = set(k, value)

  def setFeatureDim(value : Int) : this.type = set(featureDim, value)

  def setMiniBatchSize(value : Int) : this.type = set(miniBatchSize, value)

  def setMaxIter(value : Int) : this.type = set(maxIter, value)

  def setTol(value : Double) : this.type = set(tol, value)

  def setLocalInitMethod(value : String) : this.type = set(localInitMethod, value)

  def setHalflifeTime(value : Int) : this.type = set(halflifeTime, value)

  def setEpochs(value : Int) : this.type = set(epochs, value)

  def setInitMethod(value : String) : this.type = set(initMethod, value)

  def setCenterFactor(value : String) : this.type = set(centerFactor, value)

  def setCenterFactor(value : Int) : this.type = set(centerFactor, value.toString)

  def setTrialNum(value : String) : this.type = set(trialNum, value)

  def setTrialNum(value : Int) : this.type = set(trialNum, value.toString)

  def setEvalMetrics(value : String) : this.type = set(evalMetrics, "[" + value + "]")

  // Check whether required hyper-parameters are set
  override def transformSchema(schema: StructType): StructType = {
    $(featureDim)
    $(k)
    $(miniBatchSize)
    super.transformSchema(schema)
  }