25 lines
521 B
Python
25 lines
521 B
Python
|
import tensorflow as tf
|
||
|
|
import tensorflow.keras.layers as l
|
||
|
|
|
||
|
|
import generic as g
|
||
|
|
|
||
|
|
|
||
|
|
i = l.Input(shape = (2,))
|
||
|
|
h1 = l.Dense(100, activation='tanh')(i)
|
||
|
|
h2 = l.Dense(100, activation='tanh')(h1)
|
||
|
|
h3 = l.Dense(100, activation='tanh')(h2)
|
||
|
|
h4 = l.Dense(100, activation='tanh')(h3)
|
||
|
|
h5 = l.Dense(100, activation='tanh')(h4)
|
||
|
|
h6 = l.Dense(100, activation='tanh')(h5)
|
||
|
|
|
||
|
|
y = l.Dense(1)(h6)
|
||
|
|
co = l.Dense(1)(i)
|
||
|
|
|
||
|
|
o = l.Add()([y, co])
|
||
|
|
|
||
|
|
|
||
|
|
m = tf.keras.models.Model(inputs = i, outputs = o)
|
||
|
|
|
||
|
|
#g.train_generic(m, "c3")
|
||
|
|
g.verify_generic(m, "c3")
|