def get_synthetic_data():
n_samples = 5000
n_actions = 3
context_dim = 5
sigma = 1.0
samples = sample_synthetic(n_samples, n_actions, context_dim, sigma)
return n_samples, n_actions, context_dim, samples
def synthetic_data():
n_samples = 1000
n_actions = 5
context_dim = 10
samples = sample_synthetic(n_samples, n_actions, context_dim)
return n_samples, n_actions, samples