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")