diff --git a/psl-core/src/main/java/org/linqs/psl/application/learning/weight/gradient/GradientDescent.java b/psl-core/src/main/java/org/linqs/psl/application/learning/weight/gradient/GradientDescent.java index 660fa897d..6d185aca9 100644 --- a/psl-core/src/main/java/org/linqs/psl/application/learning/weight/gradient/GradientDescent.java +++ b/psl-core/src/main/java/org/linqs/psl/application/learning/weight/gradient/GradientDescent.java @@ -302,7 +302,7 @@ protected void doLearn() { protected void runMAPEvaluation() { // Compute the MAP state before evaluating so variables have assigned values. - log.trace("Running Inference."); + log.trace("Running MAP Inference."); computeMAPStateWithWarmStart(trainInferenceApplication, trainMAPTermState, trainMAPAtomValueState); evaluation.compute(trainingMap); diff --git a/psl-core/src/main/java/org/linqs/psl/application/learning/weight/gradient/optimalvalue/OptimalValue.java b/psl-core/src/main/java/org/linqs/psl/application/learning/weight/gradient/optimalvalue/OptimalValue.java index eb66d9410..302c47260 100644 --- a/psl-core/src/main/java/org/linqs/psl/application/learning/weight/gradient/optimalvalue/OptimalValue.java +++ b/psl-core/src/main/java/org/linqs/psl/application/learning/weight/gradient/optimalvalue/OptimalValue.java @@ -20,7 +20,6 @@ import org.linqs.psl.application.learning.weight.gradient.GradientDescent; import org.linqs.psl.database.AtomStore; import org.linqs.psl.database.Database; -import org.linqs.psl.model.atom.GroundAtom; import org.linqs.psl.model.atom.ObservedAtom; import org.linqs.psl.model.atom.RandomVariableAtom; import org.linqs.psl.model.rule.Rule; @@ -79,6 +78,7 @@ protected void postInitGroundModel() { protected void computeLatentInferenceIncompatibility() { fixLabeledRandomVariables(); + log.trace("Running Latent Inference."); computeMAPStateWithWarmStart(trainInferenceApplication, latentInferenceTermState, latentInferenceAtomValueState); inTrainingMAPState = true;