import matplotlib.pyplot as plt
import pandas as pd
fig, ax1 = plt.subplots()
df = pd.read_csv('./train_loss.csv')
df_fde = pd.read_csv('./best_fde.csv')
step = df["Step"].values
train_loss = df["Value"].values
step2 = df_fde["Step"].values
best_ade = df_fde["Value"].values
color = 'tab:red'
ax1.set_xlabel('epoch')
ax1.set_ylabel('Trian Loss', color=color, fontdict={'size': 12})
ax1.tick_params(axis='y', labelcolor=color)
a = ax1.plot(step, train_loss, c='red', label="train_loss")
ax2 = ax1.twinx()
color = 'tab:blue'
ax2.set_ylabel('Best Fde', color=color, fontdict={'size': 12})
b = ax2.plot(step2, best_ade, color=color, label="best_fde", linestyle="--")
ax2.tick_params(axis='y', labelcolor=color)
plt.scatter(step, train_loss, c='red', s = 5)
plt.scatter(step2, best_ade, c='blue', s = 5, marker = "s")
c = a + b
labs = [l.get_label() for l in c]
ax1.legend(c, labs, loc=0,fontsize=12)
plt.grid(True, linestyle='--', alpha=0.5)
plt.xlabel("epoch", fontdict={'size': 16})
plt.show()
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)