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