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.
|
- using Microsoft.VisualStudio.TestTools.UnitTesting;
- using System;
- using System.Collections.Generic;
- using System.Text;
- using Tensorflow.Keras.Engine;
- using Tensorflow.Keras.Layers;
- using NumSharp;
- using Tensorflow.UnitTest;
- using static Tensorflow.Binding;
-
- namespace TensorFlowNET.UnitTest.Keras
- {
- /// <summary>
- /// https://www.tensorflow.org/guide/keras/save_and_serialize
- /// </summary>
- [TestClass]
- public class ModelSaveTest : EagerModeTestBase
- {
- [TestMethod]
- public void SaveAndLoadTest()
- {
- var model = GetModel();
- }
-
- Model GetModel()
- {
- var keras = tf.keras;
-
- // Create a simple model.
- var inputs = keras.Input(shape: 32);
- var dense_layer = keras.layers.Dense(1);
- var outputs = dense_layer.Apply(inputs);
- var model = keras.Model(inputs, outputs);
- model.compile("adam", "mean_squared_error");
- return model;
- }
- }
- }
|