import os

os.environ["KERAS_BACKEND"] = "tensorflow"
os.environ["TF_CPP_MIN_LOG_LEVEL"] = "2"

import keras
import tensorflow as tf


keras.utils.set_random_seed(31)

features = tf.constant(
    [
        [0.0, 0.0, 0.1],
        [0.0, 1.0, 0.3],
        [1.0, 0.0, 0.6],
        [1.0, 1.0, 0.9],
        [0.2, 0.8, 0.4],
        [0.8, 0.2, 0.7],
        [0.3, 0.4, 0.2],
        [0.9, 0.7, 0.8],
    ],
    dtype=tf.float32,
)
labels = tf.constant(
    [[0.0], [1.0], [1.0], [0.0], [1.0], [1.0], [0.0], [0.0]],
    dtype=tf.float32,
)