Update the categorical evaluator to get observed as well as unobserved atoms.
@@ -113,7 +113,7 @@ public class CategoricalEvaluator extends Evaluator {
Set<GroundAtom> predictedCategories = getPredictedCategories(trainingMap, predicate);
- for (GroundAtom truthAtom : trainingMap.getLabelMap().values()) {
+ for (GroundAtom truthAtom : trainingMap.getAllTruths()) {
if (predicate != null && truthAtom.getPredicate() != predicate) {
continue;
}
@@ -169,7 +169,7 @@ public class CategoricalEvaluator extends Evaluator {
// or another Map<Constant, Object>, and so on.
Map<Constant, Object> predictedCategories = null;
- for (GroundAtom atom : trainingMap.getAllPredictions()) {
+ for (GroundAtom atom : trainingMap.getAllTargets()) {
if (atom.getPredicate() != predicate) {
continue;
}
