diff --git a/bdi-jade-extensions/src/bdi4jade/extension/planselection/learningbased/LearningAlgorithm.java b/bdi-jade-extensions/src/bdi4jade/extension/planselection/learningbased/LearningAlgorithm.java
index 9d21c0f..94ea388 100644
--- a/bdi-jade-extensions/src/bdi4jade/extension/planselection/learningbased/LearningAlgorithm.java
+++ b/bdi-jade-extensions/src/bdi4jade/extension/planselection/learningbased/LearningAlgorithm.java
@@ -80,17 +80,18 @@ public class LearningAlgorithm {
try {
prediction = planMetadata.getModel().classifyInstance(instance);
+ switch (planMetadata.getOptimizationFunction()) {
+ case MINIMIZE:
+ return 1 - prediction;
+ default:
+ return prediction;
+ }
} catch (Exception e) {
e.printStackTrace();
}
}
- switch (planMetadata.getOptimizationFunction()) {
- case MINIMIZE:
- return 1 - prediction;
- default:
- return prediction;
- }
+ return prediction;
}
/**
