inputs = tf.keras.Input(shape=(4,), name="features") hidden = tf.keras.layers.Dense(8, activation="relu")(inputs) scores = tf.keras.layers.Dense(1, activation="sigmoid")(hidden) model = tf.keras.Model(inputs=inputs, outputs=scores, name="number_classifier") model.compile(optimizer="adam", loss="binary_crossentropy") model.fit(features, labels, epochs=12, verbose=0)