protected void finalizeCounts() {
negIndex = labelIndex.indexOf(negLabel);
int numClasses = labelIndex.size();
if (tpCount == null || tpCount.length != numClasses) {
tpCount = new int[numClasses];
}
if (fpCount == null || fpCount.length != numClasses) {
fpCount = new int[numClasses];
}
if (fnCount == null || fnCount.length != numClasses) {
fnCount = new int[numClasses];
}
for (int i = 0; i < numClasses; i++) {
L label = labelIndex.get(i);
tpCount[i] = correctGuesses.getIntCount(label);
fnCount[i] = foundCorrect.getIntCount(label) - tpCount[i];
fpCount[i] = foundGuessed.getIntCount(label) - tpCount[i];
}
}
public int getRelevant(L label) {
return foundCorrect.getIntCount(label);
}
public int getRetrieved(L label) {
return foundGuessed.getIntCount(label);
}
public int getCorrect(L label) {
return correctGuesses.getIntCount(label);
}
