Update tensorflow_model.py

linqs · Apr 27, 2024 · 0d13a63 · 0d13a63
1 parent e81b88e
commit 0d13a63
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/psl-python/tests/resources/models/deeppsl/sign/tensorflow_model.py b/psl-python/tests/resources/models/deeppsl/sign/tensorflow_model.py
@@ -82,7 +82,7 @@ def _prepare_data(self, data):
         if len(data) == 2:
             return data
 
-        return numpy.array([numpy.asarray(data[:, :-1]), numpy.asarray([[1, 0] if label == 0 else [0, 1] for label in data[:, -1]])])
+        return numpy.array([numpy.array(data[:, :-1]), numpy.array([[1, 0] if label == 0 else [0, 1] for label in data[:, -1]])])
 
 
 def calculate_metrics(y_pred, y_truth, metrics):