import tensorflow as tf import tensorflow.keras.layers as l import generic as g m = tf.keras.models.Sequential([ l.Input(shape = (2,)), l.Dense(5, activation='relu'), l.Dense(1) ]) #g.train_generic(m, "ff") g.verify_generic(m, "ff")