Java QuantileExtraTrees.learnTreesの例

コード例 #1

ファイル: NATests.java プロジェクト: jaak-s/extraTrees

  @Test
  public void testQuantileNALearn() {
    int ndim = 5;
    QuantileExtraTrees et = getQET(100, ndim, false);
    et.learnTrees(3, 3, 5);

    double[] x = new double[ndim];
    for (int i = 0; i < x.length; i++) {
      x[i] = Double.NaN;
    }
    double[] val;
    val = et.getQuantiles(new Matrix(x, 1, ndim), 0.5);
    assertTrue(Double.isNaN(val[0]));
  }