保存训练过程中的History信息

• 保存训练的日志文件

  import pandas as pd
  import matplotlib.pyplot as plt
  # 保存训练的日志文件
  pd.DataFrame(history.history).to_csv('training_log.csv', index=False)
  

  1

• 读取保存后的训练日志文件

# 读取保存后的训练日志文件
df = pd.read_csv('training_log.csv')

# 画训练曲线
def plot_learning_curves(df):
    df.plot(figsize=(8, 5))
   plt.grid(True)
     plt.gca().set_ylim(0, 1)
    plt.show()

plot_learning_curves(df)