import matplotlib.pyplot as plt
import numpy as np
import cPickle

stuff = cPickle.load( open( "thing.pkl", "rb" ) )

fig = plt.figure()

ax1 = fig.add_subplot(121)
ax1.set_title('Entire Grid Search')
ax1.set_xlabel('Alpha [0.0,1.0,0.05]')
ax1.set_ylabel('Discount [0.0,1.0,0.05]')
ax1.set_yticks([], [])
ax1.set_xticks([], [])
ax1.imshow(stuff, cmap='hot', interpolation='nearest')

ax2 = fig.add_subplot(122)
ax2.set_title('Zoomed In to [5:,5:]')
ax2.set_yticks([], [])
ax2.set_xticks([], [])
ax2.set_xlabel('Alpha [0.0,1.0,0.05]')
ax2.set_ylabel('Discount [0.0,1.0,0.05]')
ax2.imshow(stuff[5:,5:], cmap='hot', interpolation='nearest')

plt.show()