Get JSON object before reference to result key.

linqs · Sep 30, 2023 · bda59ed · bda59ed
1 parent c62ef71
commit bda59ed
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/psl-core/src/main/java/org/linqs/psl/model/deep/DeepModel.java b/psl-core/src/main/java/org/linqs/psl/model/deep/DeepModel.java
@@ -196,7 +196,7 @@ public boolean isEpochComplete() {
         String resultString = getResultString(response);
         log.debug("Epoch end deep model results for {} : {}", this, resultString);
 
-        return response.getBoolean("is_epoch_complete");
+        return response.getJSONObject("result").getBoolean("is_epoch_complete");
     }
 
     public void predictDeepModel(Boolean learning) {
@@ -240,8 +240,8 @@ public float evalDeepModel() {
 
         String resultString = getResultString(response);
         log.debug("Eval deep model result for {} : {}", this, resultString);
-
-        return response.getFloat("loss");
+        
+        return response.getJSONObject("result").getFloat("loss");
     }
 
     public void saveDeepModel() {