tensorflowhelper-sample/LoadSaveExample.py at master · BobbyL2k/tensorflowhelper-sample · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
"""Example script for using Life"""
import numpy as np

import tensorflow as tf
import tensorflowhelper as tfh

def main():
    """Entry point function"""
    adder = tf.Variable(10.)
    adder2 = tf.Variable(7.)

    nn1 = tfh.NeuralNetwork(
        name="Inner_NN",
        layers=[
            tfh.ValidationLayer(name="Input_Validation", shape=[None], dtype=tf.float32),
            tfh.OpLayer(lambda input: input + adder, [adder]),
            tfh.OpLayer(tf.nn.relu),
            tfh.ValidationLayer(shape=[None], dtype=tf.float32)])
    nn2 = tfh.NeuralNetwork(
        name="Inner2_NN",
        layers=[
            tfh.ValidationLayer(name="Input_Validation", shape=[None], dtype=tf.float32),
            tfh.OpLayer(lambda input: input + adder2, [adder2]),
            tfh.ValidationLayer(shape=[None], dtype=tf.float32)])

    life = tfh.Life(
        tfh.NeuralNetwork(
            name="Main_NN",
            layers=[nn1, nn2]),
        optimizer=tf.train.AdamOptimizer(0.3))

    input_x = np.array([
        0, 1, 2, 3, 4
    ], dtype=np.float32)

    expect_y = np.array([
        2, 3, 4, 5, 6
    ], dtype=np.float32)

    life.connect_neural_network(sample_input=input_x, sample_output=expect_y, will_train=True)

    life.init_var()
    # life.init_network([nn1, nn2])
    # life.load_saved_model("test.ckpt", nn2)

    hypo = life.feed(input_x)

    print(hypo)

    life.save_current_model("test.ckpt", nn1)

if __name__ == "__main__":
    try:
        main()
    except tfh.utilities.TFHError as tfh_error:
        print(tfh_error)