Fixed a budget setting issue for MPPLE.
@@ -197,9 +197,12 @@ public abstract class WeightLearningApplication implements ModelApplication {
if (reasoner instanceof ADMMReasoner) {
int maxIterations = config.getInt(ADMMReasoner.MAX_ITER_KEY, ADMMReasoner.MAX_ITER_DEFAULT);
((ADMMReasoner)reasoner).setMaxIter((int)Math.ceil(maxIterations * budget));
+
+ if (termStore instanceof ADMMTermStore) {
((ADMMTermStore)termStore).resetLocalVairables();
}
}
+ }
/**
* Pass in all the ground model infrastructure.
