Package edu.uci.jforestsx.dataset

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
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
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

Related Classes of edu.uci.jforestsx.dataset.RankingDataset

  • edu.uci.jforestsx.applications.RankingApp
  • edu.uci.jforestsx.applications.Runner
    • Copyright © 2018 www.massapicom. 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.