def main(fname, means, rnd_seed=1):
random.seed(rnd_seed)
n_arms = len(means)
random.shuffle(means)
arms = list(map(lambda mu: BernoulliArm(mu), means))
print("Best arm is " + str(Solver.ind_max(means)))
f = open(fname, "w")
for temperature in [0.1, 0.2, 0.3, 0.4, 0.5]:
algo = Softmax(temperature, [], [])
algo.initialize(n_arms)
results = test_algorithm(algo, arms, 5000, 250)
for i in range(len(results[0])):
f.write(str(temperature) + "\t")
f.write(
"\t".join([str(results[j][i])
for j in range(len(results))]) + "\n")
f.close()
