import os

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

import tensorflow as tf

tf.keras.utils.set_random_seed(7)

train_features = tf.constant(
    [
        [510.0, 6.5, 1.2],
        [620.0, 7.1, 0.8],
        [470.0, 5.9, 1.7],
        [730.0, 8.0, 0.4],
        [690.0, 7.8, 0.6],
        [540.0, 6.8, 1.0],
        [455.0, 5.7, 1.9],
        [760.0, 8.3, 0.3],
    ],
    dtype=tf.float32,
)
train_labels = tf.constant([0, 1, 0, 1, 1, 0, 0, 1], dtype=tf.float32)

validation_features = tf.constant(
    [
        [500.0, 6.4, 1.1],
        [710.0, 7.9, 0.5],
        [480.0, 6.0, 1.6],
        [745.0, 8.1, 0.4],
    ],
    dtype=tf.float32,
)
validation_labels = tf.constant([0, 1, 0, 1], dtype=tf.float32)

feature_scale = tf.constant([1000.0, 10.0, 10.0], dtype=tf.float32)