brute_force_parallelism

brute_force_parallelism.java

CrossValidation<DecisionTreeNode, Integer, Vector> scoreCalculator
                    = new CrossValidation<>();

ParamGrid paramGrid = new ParamGrid()
  .withParameterSearchStrategy(new BruteForceStrategy())
  .addHyperParam("p", normalizationTrainer::withP, new Double[] {1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0})
  .addHyperParam("maxDeep", trainerCV::withMaxDeep, new Double[] {1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0})
  .addHyperParam("minImpurityDecrease", trainerCV::withMinImpurityDecrease, new Double[] {0.0, 0.1, 0.2, 0.3, 0.4, 0.5, 0.6, 0.7, 0.8, 0.9, 1.0});

scoreCalculator
  .withIgnite(ignite)
  .withUpstreamCache(dataCache)
  .withEnvironmentBuilder(LearningEnvironmentBuilder.defaultBuilder()
                          .withParallelismStrategyTypeDependency(ParallelismStrategy.ON_DEFAULT_POOL) // <======= set up the ParallelismStrategy
                          .withLoggingFactoryDependency(ConsoleLogger.Factory.LOW))
  .withTrainer(trainerCV)
  .isRunningOnPipeline(false)
  .withMetric(MetricName.ACCURACY)
  .withFilter(split.getTrainFilter())
  .withPreprocessor(normalizationPreprocessor)
  .withAmountOfFolds(3)
  .withParamGrid(paramGrid)