def create_data(datatype, n=1000):
x_train, y_train, _ = generate_data(n=n, datatype=datatype, seed=0)
x_val, y_val, datatypes_val = generate_data(n=100,
datatype=datatype,
seed=1)
input_shape = x_train.shape[1]
return x_train, y_train, x_val, y_val, datatypes_val, input_shape
def create_data(datatype, n=1000):
"""
Create train and validation datasets.
"""
x_train, y_train, _ = generate_data(n=n, datatype=datatype, seed=0)
x_val, y_val, datatypes_val = generate_data(n=10**5,
datatype=datatype,
seed=1)
input_shape = x_train.shape[1]
return x_train, y_train, x_val, y_val, datatypes_val, input_shape