Browse Source

test: Added an `evaluate` method call to a unit test for a multi-input model.

pull/1260/head
Aleksej Solomatin 1 year ago
parent
commit
5142ad658c
1 changed files with 7 additions and 0 deletions
  1. +7
    -0
      test/TensorFlowNET.Keras.UnitTest/MultiInputModelTest.cs

+ 7
- 0
test/TensorFlowNET.Keras.UnitTest/MultiInputModelTest.cs View File

@@ -54,6 +54,13 @@ namespace Tensorflow.Keras.UnitTest
var x = new NDArray[] { x1, x2 }; var x = new NDArray[] { x1, x2 };
model.fit(x, dataset.Train.Labels, batch_size: 8, epochs: 3); model.fit(x, dataset.Train.Labels, batch_size: 8, epochs: 3);


x1 = x1["0:8"];
x2 = x1;

x = new NDArray[] { x1, x2 };
var y = dataset.Train.Labels["0:8"];
(model as Engine.Model).evaluate(x, y);

x1 = np.ones((1, 28, 28, 1), TF_DataType.TF_FLOAT); x1 = np.ones((1, 28, 28, 1), TF_DataType.TF_FLOAT);
x2 = np.zeros((1, 28, 28, 1), TF_DataType.TF_FLOAT); x2 = np.zeros((1, 28, 28, 1), TF_DataType.TF_FLOAT);
var pred = model.predict((x1, x2)); var pred = model.predict((x1, x2));


Loading…
Cancel
Save