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.Linq;
using static Tensorflow.Binding;
using static Tensorflow.KerasApi;

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

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