model = tf.keras.Sequential([

    tf.keras.layers.Dense(8, input_shape=(len(train_x[0]))),

    tf.keras.layers.Dense(8),

    tf.keras.layers.Dense(len(train_y[0]), activation="softmax"),

])

​

model.compile(optimizer="adam", loss="categorical_crossentropy", metrics=["accuracy"])

model.fit(train_x, train_y, epochs=epochs, batch_size=batch_size)