In [1]:
import pandas as pd
import numpy as np
%matplotlib inline

In [2]:
initial = np.load("lstm_basic/initial.npy")
initial_df = pd.DataFrame(initial, columns=["initial"])

In [3]:
output1 = np.load("lstm_basic/output.npy")
output1_df = pd.DataFrame(output1, columns=["output (optimizer: GD)"], index=range(len(initial), len(initial) + len(output1)))
losses1_df = pd.DataFrame(np.load("lstm_basic/losses.npy"), columns=["epoch", "loss (optimizer: GD)"])

In [4]:
output2 = np.load("lstm_adam/output.npy")
output2_df = pd.DataFrame(output2, columns=["output (optimizer: Adam)"], index=range(len(initial), len(initial) + len(output2)))
losses2_df = pd.DataFrame(np.load("lstm_adam/losses.npy"), columns=["epoch", "loss (optimizer: Adam)"])

In [5]:
train = np.load("train_data/normal.npy")
train_df = pd.DataFrame(train[:, 0], columns=["train"])

In [6]:
merged = pd.concat([train_df, initial_df, output1_df, output2_df])
merged.plot(figsize=(15, 5), grid=True, style=["-", "-", "-", "k--"])


Out[6]:
<matplotlib.axes._subplots.AxesSubplot at 0x104849dd8>

In [7]:
merged = pd.merge(losses1_df, losses2_df)
merged.plot(figsize=(15, 5), grid=True, logy=True, x="epoch")


Out[7]:
<matplotlib.axes._subplots.AxesSubplot at 0x1049285c0>

In [ ]: