You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.

ModelSaveTest.cs 837 B

123456789101112131415161718192021222324252627282930
  1. using Microsoft.VisualStudio.TestTools.UnitTesting;
  2. using Tensorflow.Keras.Engine;
  3. using static Tensorflow.KerasExt;
  4. namespace TensorFlowNET.UnitTest.Keras
  5. {
  6. /// <summary>
  7. /// https://www.tensorflow.org/guide/keras/save_and_serialize
  8. /// </summary>
  9. [TestClass]
  10. public class ModelSaveTest : EagerModeTestBase
  11. {
  12. [TestMethod]
  13. public void SaveAndLoadTest()
  14. {
  15. var model = GetModel();
  16. }
  17. Model GetModel()
  18. {
  19. // Create a simple model.
  20. var inputs = keras.Input(shape: 32);
  21. var dense_layer = keras.layers.Dense(1);
  22. var outputs = dense_layer.Apply(inputs);
  23. var model = keras.Model(inputs, outputs);
  24. model.compile("adam", "mean_squared_error");
  25. return model;
  26. }
  27. }
  28. }