code snippet for Hyperparameters (part II): Random Search on Spark

randomSearch_lgbm

import breeze.stats.distributions.{Gamma,Uniform,Poisson}
import com.microsoft.ml.spark.LightGBMClassifier

import tuning.RandomGridBuilder
object example_lgbm{
  def main(args: Array[String]): Unit = {

    val lgbm = new LightGBMClassifier()
    val randomGrid = new RandomGridBuilder(5)
      .addDistr(lgbm.learningRate,Gamma(1.0,0.1))
      .addDistr(lgbm.baggingFraction,Uniform(0.7,0.99))
      .addDistr(lgbm.featureFraction,Uniform(0.7,0.99))
      .addDistr(lgbm.maxDepth,Poisson(10))
      .addDistr(lgbm.numIterations,Poisson(200))
      .build()

    println(randomGrid.toList)
  }
}