Browse Source

remove UnitTest for `keras.datasets.imdb`

tags/v0.110.0-LSTM-Model
lingbai-kong 2 years ago
parent
commit
c23b24633f
1 changed files with 0 additions and 15 deletions
  1. +0
    -15
      test/TensorFlowNET.UnitTest/Dataset/DatasetTest.cs

+ 0
- 15
test/TensorFlowNET.UnitTest/Dataset/DatasetTest.cs View File

@@ -2,7 +2,6 @@
using System; using System;
using System.Linq; using System.Linq;
using static Tensorflow.Binding; using static Tensorflow.Binding;
using static Tensorflow.KerasApi;


namespace TensorFlowNET.UnitTest.Dataset namespace TensorFlowNET.UnitTest.Dataset
{ {
@@ -196,19 +195,5 @@ namespace TensorFlowNET.UnitTest.Dataset


Assert.IsFalse(allEqual); Assert.IsFalse(allEqual);
} }
[TestMethod]
public void GetData()
{
var vocab_size = 20000;
var dataset = keras.datasets.imdb.load_data(num_words: vocab_size);
var x_train = dataset.Train.Item1;
Assert.AreEqual(x_train.dims[0], 25000);
var y_train = dataset.Train.Item2;
Assert.AreEqual(y_train.dims[0], 25000);
var x_val = dataset.Test.Item1;
Assert.AreEqual(x_val.dims[0], 25000);
var y_val = dataset.Test.Item2;
Assert.AreEqual(y_val.dims[0], 25000);
}
} }
} }

Loading…
Cancel
Save