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