(incubator-wayang) branch rel/1.0.0-rc5 updated: updating Tensorflow versioning

[zkaoudi]

This is an automated email from the ASF dual-hosted git repository.

zkaoudi pushed a commit to branch rel/1.0.0-rc5
in repository https://gitbox.apache.org/repos/asf/incubator-wayang.git


The following commit(s) were added to refs/heads/rel/1.0.0-rc5 by this push:
     new b2ec2807 updating Tensorflow versioning
b2ec2807 is described below

commit b2ec2807ad510ef6136881aa80095f6800b4c57e
Author: Zoi Kaoudi <zkao...@users.noreply.github.com>
AuthorDate: Fri Feb 21 14:09:19 2025 +0100

    updating Tensorflow versioning
---
 .../apache/wayang/tensorflow/model/TensorflowModel.java | 17 ++++++-----------
 1 file changed, 6 insertions(+), 11 deletions(-)

diff --git 
a/wayang-platforms/wayang-tensorflow/src/main/java/org/apache/wayang/tensorflow/model/TensorflowModel.java
 
b/wayang-platforms/wayang-tensorflow/src/main/java/org/apache/wayang/tensorflow/model/TensorflowModel.java
index 80113ffb..e881f95d 100644
--- 
a/wayang-platforms/wayang-tensorflow/src/main/java/org/apache/wayang/tensorflow/model/TensorflowModel.java
+++ 
b/wayang-platforms/wayang-tensorflow/src/main/java/org/apache/wayang/tensorflow/model/TensorflowModel.java
@@ -22,10 +22,7 @@ import org.apache.wayang.basic.model.DLModel;
 import org.apache.wayang.basic.model.op.Input;
 import org.apache.wayang.basic.model.op.Op;
 import org.apache.wayang.basic.model.optimizer.Optimizer;
-import org.tensorflow.Graph;
-import org.tensorflow.Operand;
-import org.tensorflow.Session;
-import org.tensorflow.Tensor;
+import org.tensorflow.*;
 import org.tensorflow.ndarray.*;
 import org.tensorflow.ndarray.index.Indices;
 import org.tensorflow.op.Ops;
@@ -122,14 +119,12 @@ public class TensorflowModel extends DLModel implements 
AutoCloseable {
                     if (accuracyCalculation != null) {
                         runner.fetch(accuracyCalculation.getName());
                     }
-                    List<Tensor> ret = runner.run();
-                    try (TFloat32 loss = (TFloat32) ret.get(0)) {
+                    try (Result ret = runner.run()) {
+                        TFloat32 loss = (TFloat32) ret.get(0);
                         System.out.printf("[epoch %d, batch %d] loss: %f ", i 
+ 1, start / batchSize + 1, loss.getFloat());
-                    }
-                    if (accuracyCalculation != null) {
-                        try (TFloat32 acc = (TFloat32) ret.get(1)) {
-                            System.out.printf("accuracy: %f ", acc.getFloat());
-                        }
+
+                        TFloat32 acc = (TFloat32) ret.get(1);
+                        System.out.printf("accuracy: %f ", acc.getFloat());
                     }
                     System.out.println();
                 }