diff --git a/test/TensorFlowNET.Keras.UnitTest/Layers/LayersTest.cs b/test/TensorFlowNET.Keras.UnitTest/Layers/LayersTest.cs index 63e959f5..ba54a0f4 100644 --- a/test/TensorFlowNET.Keras.UnitTest/Layers/LayersTest.cs +++ b/test/TensorFlowNET.Keras.UnitTest/Layers/LayersTest.cs @@ -40,7 +40,7 @@ namespace TensorFlowNET.Keras.UnitTest /// /// Custom layer test, used in Dueling DQN /// - [TestMethod] + [TestMethod, Ignore] public void TensorFlowOpLayer() { var layers = keras.layers; @@ -58,6 +58,9 @@ namespace TensorFlowNET.Keras.UnitTest metrics: new[] { "acc" }); model.summary(); Assert.AreEqual(model.Layers.Count, 8); + var result = model.predict(tf.constant(np.arange(24).astype(np.float32)[np.newaxis, Slice.All])); + Assert.AreEqual(result.shape, new TensorShape(1, 24)); + model.fit(np.arange(24).astype(np.float32)[np.newaxis, Slice.All], np.arange(24).astype(np.float32)[np.newaxis, Slice.All], verbose: 0); } ///