java - H2O 中的 GridSearch API

Question

我正在尝试使用 GridSearch Scala 中的 api H2O 。我找到了this文档显示了在 R 和 Python 中做什么，但 Java 文档在实际获得最佳模型之前就停止了。谁能告诉我最后几行？ IE 给出 Grid<MP>如何获得最佳超参数组合或最佳模型？

编辑

这是我的代码到目前为止的样子(它是在 Scala 中，但显然我是在 H2O 中调用 Java 代码)。

private[this] def hyperopt(hyperParams: Map[String, Array[Object]]): M = {
  val search = GridSearch.startGridSearch(
    null,
    params,
    hyperParams.asJava,
    new SimpleParametersBuilderFactory[P],
    new HyperSpaceSearchCriteria.RandomDiscreteValueSearchCriteria)
  val grid = search.get().asInstanceOf[Grid[P]]
  val scoringInfo = grid.getScoringInfos.last // This is best scoringInfo cause it gets sorted along the way
  //val bestParams = scoringInfo.???
  //grid.getModel(bestParams).asInstanceOf[M]
}

关键问题是我不知道如何写最后两行。

经过一番挖掘，我想出了这个解决方案，虽然我不喜欢，但确实有效

private[this] def hyperopt(hyperParams: (Map[String, Array[Object]], String)): M = {
  val search = GridSearch.startGridSearch(
    null,
    params,
    hyperParams._1.asJava,
    new SimpleParametersBuilderFactory[P],
    new HyperSpaceSearchCriteria.RandomDiscreteValueSearchCriteria)
  val grid = search.get()
  val summary = grid.createSummaryTable(grid.getModelKeys, hyperParams._2, true)

  info("Hyper parameter results")
  info(summary.toString)

  // TODO Fix this hack.  I don't know a better way to do this now.
  val bestModelId = summary.get(0, 1)
  grid.getModels.find(m => m._key.toString == bestModelId).get.asInstanceOf[M]
}

Answer 1

如果没记错的话，Grid grid = (Grid) gs.get()，会返回一个完整的网格。现在，什么构成了最佳模型？如果您说这是最高的 ROC，则按“auc”降序对列表进行排序并选择第一个元素 - 与 this post 非常相似.