Skip to content

Commit

Permalink
added comment
Browse files Browse the repository at this point in the history
  • Loading branch information
leokim-l committed Nov 26, 2024
1 parent c9f7b2c commit 635e96c
Showing 1 changed file with 9 additions and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -90,6 +90,14 @@ public static void outputCorrectPPKt(List<CorrectResult> correctResultList) {
outputCorrectResults(correctResultList, PROMPT_DIR);
}

/**
* Output correct results to file.
* @param correctResultList an ArrayList of the correct results, coming from outputPromptsEnglish
* or outputPromptsEnglishFromIndividuals. Format is a list of triples
* promptFileName, Disease ID, Disease Label
* @param basename the directory where prompts are output to.
* @return The square root of the given number.
*/
public static void outputCorrectResults(List<CorrectResult> correctResultList, String basename) {
File outfile = new File(basename + File.separator + "correct_results.tsv");
try (BufferedWriter bw = new BufferedWriter(new FileWriter(outfile))) {
Expand All @@ -99,7 +107,7 @@ public static void outputCorrectResults(List<CorrectResult> correctResultList,
} catch (IOException e) {
e.printStackTrace();
}
System.out.printf("[INFO] Output a total of %d prompts in en, es, nl, de, tr, zh and it.\n", correctResultList.size());
System.out.printf("[INFO] Output a total of %d prompts in en, es, nl, cs, de, tr, zh and it.\n", correctResultList.size());
}


Expand Down

0 comments on commit 635e96c

Please sign in to comment.