Skip to content

Commit 3025891

Browse files
committed
fixed bug
1 parent 862cc22 commit 3025891

File tree

3 files changed

+5
-4
lines changed

3 files changed

+5
-4
lines changed

src/processing/BM25Calculator.java

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -318,7 +318,8 @@ public static BookmarkReader predictSample(String filename, int trainSize, int s
318318
for (int i = 0; i < cfValues.size(); i++) {
319319
Map<Integer, Double> modelVal = cfValues.get(i);
320320
predictionValues.add(Ints.toArray(modelVal.keySet()));
321-
System.out.println(modelVal.values().toString());
321+
// just for debugging
322+
//System.out.println(modelVal.values().toString());
322323
}
323324
String suffix = "_cf_";
324325
if (!userBased) {

src/processing/MetricsCalculator.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -90,7 +90,7 @@ public MetricsCalculator(PredictionFileReader reader, String outputFile, int k,
9090
if (wikiReader != null) {
9191
trainSize = this.wikiReader.getCountLimit();
9292
// TODO: could be replaced by tags
93-
resourceTopics = Utilities.getUniqueTopicMaps(wikiReader.getBookmarks().subList(0, trainSize), true);
93+
resourceTopics = Utilities.getUniqueTopicMaps(wikiReader.getBookmarks(), true);
9494
//resourceTopics = Utilities.getRelativeTagMaps(wikiReader.getBookmarks().subList(0, trainSize), true);
9595
}
9696
for (PredictionData data : this.reader.getPredictionData()) {

src/test/Pipeline.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -86,8 +86,8 @@ public static void main(String[] args) {
8686
String dir = DATASET + "_core";
8787
String path = dir + "/" + DATASET + "_sample";
8888
//getStatistics(path);
89-
//startCfResourceCalculator("del_core", "del_core/del_sample", 1, 20, true, false, false, false, Features.ENTITIES);
90-
//startResourceCIRTTCalculator("bib_core", "bib_core/bib_sample", "", 1, 20, Features.ENTITIES, false, true, false, true);
89+
//startCfResourceCalculator(dir, path, 1, 20, true, false, false, false, Features.ENTITIES);
90+
//startResourceCIRTTCalculator(dir, path, "", 1, 20, Features.ENTITIES, false, true, false, true);
9191
//startBaselineCalculatorForResources(dir, path, 1, false);
9292

9393

0 commit comments

Comments
 (0)