import os

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

import tensorflow as tf

tf.get_logger().setLevel("ERROR")
tf.keras.utils.set_random_seed(7)

features = tf.random.stateless_normal((512, 4), seed=(7, 11))
scores = 1.5 * features[:, 0] - features[:, 1] + 0.5 * features[:, 2]
labels = tf.cast(scores > 0, tf.float32)[:, None]

train_dataset = (
    tf.data.Dataset.from_tensor_slices((features[:448], labels[:448]))
    .shuffle(448, seed=7, reshuffle_each_iteration=True)
    .batch(32)
)
validation_dataset = tf.data.Dataset.from_tensor_slices(
    (features[448:], labels[448:])
).batch(32)