plot_overlap.py

import numpy as np
import matplotlib.pyplot as plt

resume = './baseline/vgg16-cifar100/'

path_30 = resume + 'overlap_0.3.npy'
path_50 = resume + 'overlap_0.5.npy'
path_70 = resume + 'overlap_0.7.npy'

overlap_50 = np.load(path_50)
overlap_30 = np.load(path_30)

# print(overlap[100, :])
fig = plt.figure('overlap', figsize=(20,10))
ax1 = fig.add_subplot(1,3,1)
overlap = ax1.imshow(overlap_30, cmap=plt.cm.viridis, interpolation='none', vmin=0, vmax=1)
cb = plt.colorbar(overlap, fraction=0.046, pad=0.04)
ax1.set_title('Correlation map (p=0.3)', fontsize=20)
ax1.set_xlabel('epochs', fontsize=15)
ax1.set_ylabel('epochs', fontsize=15)
ax2 = fig.add_subplot(1,3,2)
overlap = ax2.imshow(overlap_50, cmap=plt.cm.viridis, interpolation='none', vmin=0, vmax=1)
cb = plt.colorbar(overlap, fraction=0.046, pad=0.04)
ax2.set_title('Correlation map (p=0.5)', fontsize=20)
ax2.set_xlabel('epochs', fontsize=15)
ax2.set_ylabel('epochs', fontsize=15)
ax3 = fig.add_subplot(1,3,3)
overlap = ax3.imshow(overlap_70, cmap=plt.cm.viridis, interpolation='none', vmin=0, vmax=1)
cb = plt.colorbar(overlap, fraction=0.046, pad=0.04)
ax3.set_title('Correlation map (p=0.7)', fontsize=20)
ax3.set_xlabel('epochs', fontsize=15)
ax3.set_ylabel('epochs', fontsize=15)

plt.savefig('vgg16-cifar100.png')