20 lines
315 B
Python
20 lines
315 B
Python
|
import tensorflow as tf
|
||
|
|
import tensorflow.keras.layers as l
|
||
|
|
|
||
|
|
import generic as g
|
||
|
|
|
||
|
|
|
||
|
|
i = l.Input(shape = (2,))
|
||
|
|
h1 = l.Dense(20, activation='relu')(i)
|
||
|
|
|
||
|
|
y = l.Dense(1)(h1)
|
||
|
|
co = l.Dense(1)(i)
|
||
|
|
|
||
|
|
o = l.Add()([y, co])
|
||
|
|
|
||
|
|
|
||
|
|
m = tf.keras.models.Model(inputs = i, outputs = o)
|
||
|
|
|
||
|
|
#g.train_generic(m, "c1")
|
||
|
|
g.verify_generic(m, "c1")
|