hill_climbing_plot.py

import numpy as np
import matplotlib.pyplot as plt
from assignment1 import *

NUM_TRIALS = 100
SIZE = 5
NUM_ITERATIONS = 1000
FILENAME = "hill_climb_{}_{}x{}.gz".format(SIZE,NUM_TRIALS,NUM_ITERATIONS)

data = []
for i in range(NUM_TRIALS):
    _, vals = hill_climb(generate_puzzle(SIZE),NUM_ITERATIONS)
    data.append(vals)
data = np.array(data)
np.savetxt(FILENAME,data)
plt.plot([np.mean(r) for r in data.T])
plt.show()