Skip to content

Commit

Permalink
Using expertSetParam and expertGetParam
Browse files Browse the repository at this point in the history
  • Loading branch information
yu-iskw committed Jul 17, 2015
1 parent c8dc6e6 commit effc650
Showing 1 changed file with 4 additions and 4 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -78,7 +78,7 @@ private[clustering] trait KMeansParams
final val initMode = new Param[String](this, "initMode", "initialization algorithm",
(value: String) => MLlibKMeans.validateInitMode(value))

/** @group getExpertParam */
/** @group expertGetParam */
def getInitMode: String = $(initMode)

/**
Expand All @@ -89,7 +89,7 @@ private[clustering] trait KMeansParams
final val initSteps = new IntParam(this, "initSteps", "number of steps for k-means||",
(value: Int) => value > 0)

/** @group getExpertParam */
/** @group expertGetParam */
def getInitSteps: Int = $(initSteps)

/**
Expand Down Expand Up @@ -163,10 +163,10 @@ class KMeans(override val uid: String) extends Estimator[KMeansModel] with KMean
/** @group setParam */
def setK(value: Int): this.type = set(k, value)

/** @group setExpertParam */
/** @group expertSetParam */
def setInitMode(value: String): this.type = set(initMode, value)

/** @group setExpertParam */
/** @group expertSetParam */
def setInitSteps(value: Int): this.type = set(initSteps, value)

/** @group setParam */
Expand Down

0 comments on commit effc650

Please sign in to comment.