class SimpleMnistDataVisualizer:
def __init__(self, X_train, y_train, mapp):
self.X_train = X_train
self.y_train = y_train
self.mapp = mapp
self.plotter = Plotter()
def plot_specified_digit(self, code):
"""
Plot a digit or character
:param code:
"""
for i in range(self.y_train.shape[0]):
if (self.y_train[i, code] == 1):
self.plotter.plot_image(self.X_train[i].reshape(28,28))
break
def plot_first_digit(self, ):
"""
Plot first element of the training set
"""
self.plotter.plot_image(self.X_train[0].reshape(28, 28))
def plot_range(self):
"""
Plot characters from training set using a range
"""
for i in range(100, 109):
plt.subplot(330 + (i + 1))
plt.imshow(self.X_train[i].reshape(28, 28), cmap=plt.get_cmap('gray'))
plt.title(chr(self.mapp[np.argmax(self.y_train[i])]))
plt.show()
def plotgraph(self, epochs, acc, val_acc):
"""
Plot a graph
:param epochs:
:param acc:
:param val_acc:
"""
self.plotter.plotgraph(epochs, acc, val_acc)
def plot_loss_acc(self):
"""
Plot loss based on parameters of each epoch
"""
history_path = './experiments/2019-12-15/conv_emnist_from_config/history_params/parameters.csv'
data_frame = pd.read_csv(history_path, delimiter=',')
self.plotter.plotgraph(data_frame['epoch'].values, data_frame['acc'].values, data_frame['val_acc'])
self.plotter.plotgraph(data_frame['epoch'].values, data_frame['loss'].values, data_frame['val_loss'])
history1 = "experiments/2019-12-25/xy_emnist_from_config/history_params/parameters.csv"
history2 = "experiments/2019-12-25/xy_emnist_x_from_config/history_params/parameters.csv"
history3 = "experiments/2019-12-26/xy_emnist_bn_from_config/history_params/parameters.csv"
history_list = [history1, history2, history3]
index = 0
for history in history_list:
index += 1
file = open(history, "r")
reader = csv.reader(file)
epochs = []
acc = []
val_acc = []
loss = []
val_loss = []
for item in reader:
if reader.line_num == 1:
continue
epochs.append(int(item[0]))
acc.append(float(item[1]))
loss.append(float(item[2]))
val_acc.append(float(item[3]))
val_loss.append(float(item[4]))
Plotter.plotgraph(title="model_" + str(index),
epochs=epochs,
acc=acc,
val_acc=val_acc)
Plotter.plotlossgraph(title="model_" + str(index),
epochs=epochs,
loss=loss,
val_loss=val_loss)
