April 2017
Intermediate to advanced
532 pages
12h 39m
English
Scala
object GradientBoostedTreesIterations{ def main(args: Array[String]) { val data = GradientBoostedTreesUtil.getTrainTestData() val train_data = data._1 val test_data = data._2 val iterations_param = Array(1, 5, 10, 15, 18) val i = 0 val resultsMap = new scala.collection.mutable.HashMap[ String, String] val dataset = new DefaultCategoryDataset() for(i <- 0 until iterations_param.length) { val iteration = iterations_param(i) val rmsle = GradientBoostedTreesUtil.evaluate(train_data, test_data,iteration,maxDepth) resultsMap.put(iteration.toString,rmsle.toString) dataset.addValue(rmsle, "RMSLE", iteration) } val chart = new LineChart( "Iterations" , "GradientBoostedTrees : RMSLE vs Iterations") chart.exec("Iterations","RMSLE",dataset) ...Read now
Unlock full access