import os
from pathlib import Path

os.environ["KERAS_BACKEND"] = "jax"

import keras
import keras_tuner
import numpy as np


keras.utils.set_random_seed(17)

rng = np.random.default_rng(17)
x = rng.normal(size=(240, 4)).astype("float32")
y = (
    (0.8 * x[:, 0] - 0.4 * x[:, 1] + 0.3 * x[:, 2] + 0.1 * x[:, 3]) > 0
).astype("float32")

x_train, x_val, x_test = x[:160], x[160:200], x[200:]
y_train, y_val, y_test = y[:160], y[160:200], y[200:]