import tensorflow as tf
import tensorflow.keras.layers as l

import generic as g


i = l.Input(shape = (2,))
h1 = l.Dense(1000, 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, "c2")
g.verify_generic(m, "c2")