in sagemaker-spark-sdk/src/main/scala/com/amazonaws/services/sagemaker/sparksdk/algorithms/SageMakerAlgorithmParams.scala [30:56]
def getMiniBatchSize: Int = $(miniBatchSize)
/**
* The dimension of the input vectors. Must be > 0.
* Required.
*
*/
val featureDim : IntParam = new IntParam(this, "feature_dim",
"The dimension of the input vectors. Must be > 0.", ParamValidators.gtEq(1))
def getFeatureDim: Int = $(featureDim)
protected def autoOrAboveParamValidator(lowerBound: Double,
inclusive: Boolean): String => Boolean = {
(value: String) =>
try {
value == "auto" || {
if (inclusive) {
value.toDouble >= lowerBound
}
else {
value.toDouble > lowerBound
}
}
} catch {
case e: NumberFormatException => false
}
}