|
|
@@ -40,7 +40,7 @@ namespace TensorFlowNET.Keras.UnitTest |
|
|
|
/// <summary> |
|
|
|
/// Custom layer test, used in Dueling DQN |
|
|
|
/// </summary> |
|
|
|
[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); |
|
|
|
} |
|
|
|
|
|
|
|
/// <summary> |
|
|
|