def main(algo_name, bound_name, scenario_name, setting_name):
assert(setting_name == "bandit" or setting_name == "finite")
scenario = Scenario(scenario_name)
print("Running scenario", scenario_name, "with algorithm", algo_name, "and bound", bound_name)
num_iterations = 100#1
algo = Algorithm.make(algo_name, bound_name, scenario)
width_avg, sample_avg, correct_avg, UB_avg, mean_avg = run(scenario, algo, setting_name, algo_name, scenario_name, bound_name)
for i in range(num_iterations - 1):
algo = Algorithm.make(algo_name, bound_name, scenario)
width_history, sample_history, correct_history, UB_history, mean_history = run(scenario, algo, setting_name, algo_name, scenario_name, bound_name)
width_avg += width_history
sample_avg += sample_history
correct_avg += correct_history
UB_avg += UB_history
mean_avg += mean_history
width_avg /= num_iterations
sample_avg /= num_iterations
correct_avg /= num_iterations
UB_avg /= num_iterations
mean_avg /= num_iterations
save_data(setting_name, algo_name, scenario_name, bound_name, ["widths", "samples", "correct", "UB", "mean"], [width_avg, sample_avg, correct_avg, UB_avg, mean_avg])
