Examples of RankingDataset

Examples of edu.uci.jforestsx.dataset.RankingDataset

79
80
81
82
83
84
85
    trainingConfig.init(configHolder);
  }

  @Override
  protected Dataset newDataset() {
    return new RankingDataset();
  }
View Full Code Here

Examples of edu.uci.jforestsx.dataset.RankingDataset

87
88
89
90
91
92
93
94
95
96
  @Override
  public void initDataset(Dataset dataset) throws Exception {
    if (dataset == null || !dataset.needsInitialization) {
      return;
    }
    RankingDataset rankingDataset = (RankingDataset) dataset;
    int[][] labelCounts = NDCGEval.getLabelCountsForQueries(rankingDataset.targets, rankingDataset.queryBoundaries);
    rankingDataset.maxDCG = NDCGEval.getMaxDCGForAllQueriesUptoTruncation(rankingDataset.targets, rankingDataset.queryBoundaries,
        NDCGEval.MAX_TRUNCATION_LEVEL, labelCounts);
  }
View Full Code Here

Examples of edu.uci.jforestsx.dataset.RankingDataset

150
151
152
153
154
155
156
157
158
159
160
     * Load the data set
     */
    InputStream in = new IOUtils().getInputStream((String) options.valueOf("test-file"));
    Sample sample;
    if (options.has("ranking")) {
      RankingDataset dataset = new RankingDataset();
      RankingDatasetLoader.load(in, dataset);
      sample = new RankingSample(dataset);
    } else {
      Dataset dataset = new Dataset();
      DatasetLoader.load(in, dataset);
View Full Code Here
TOP
Copyright © 2018 www.massapi.com. All rights reserved.
All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.