快捷導(dǎo)航

在keras中實(shí)現(xiàn)查看其訓(xùn)練loss值

更新時(shí)間：2020年06月16日 10:58:51 作者：感到鴨力

這篇文章主要介紹了在keras中實(shí)現(xiàn)查看其訓(xùn)練loss值，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧

想要查看每次訓(xùn)練模型后的 loss 值變化需要如下操作

loss_value= [ ]
self.history = model.fit(state,target_f,epochs=1, batch_size =32）
b = abs(float(self.history.history[‘loss'][0]))
loss_value.append(b)
print(loss_value)
loss_value = np.array( loss_value)
x = np.array(range(len( loss_value）））
plt.plot(x, loss_value, c = ‘g')
pt.svefit('c地址‘， dpi= 100)
plt.show()

scipy.sparse 稀疏矩陣函數(shù)集合

pandas 用于在各種文件中提取，并處理分析數(shù)據(jù)；有DataFrame數(shù)據(jù)結(jié)構(gòu)，類(lèi)似表格。

x=np.linspace(-10, 10, 100) 生成100個(gè)在-10到10之間的數(shù)組

補(bǔ)充知識(shí)：對(duì)keras訓(xùn)練過(guò)程中l(wèi)oss，val_loss，以及accuracy，val_accuracy的可視化

我就廢話不多說(shuō)了，大家還是直接看代碼吧！

 hist = model.fit_generator(generator=data_generator_reg(X=x_train, Y=[y_train_a,y_train_g], batch_size=batch_size),
         steps_per_epoch=train_num // batch_size,
         validation_data=(x_test, [y_test_a,y_test_g]),
         epochs=nb_epochs, verbose=1,
         workers=8, use_multiprocessing=True,
         callbacks=callbacks)

 logging.debug("Saving weights...")
 model.save_weights(os.path.join(db_name+"_models/"+save_name, save_name+'.h5'), overwrite=True)
 pd.DataFrame(hist.history).to_hdf(os.path.join(db_name+"_models/"+save_name, 'history_'+save_name+'.h5'), "history")

在訓(xùn)練時(shí)，會(huì)輸出如下打?。?/p>

640/640 [==============================] - 35s 55ms/step - loss: 4.0216 - mean_absolute_error: 4.6525 - val_loss: 3.2888 - val_mean_absolute_error: 3.9109

有訓(xùn)練loss，訓(xùn)練預(yù)測(cè)準(zhǔn)確度，以及測(cè)試loss，以及測(cè)試準(zhǔn)確度，將文件保存后，使用下面的代碼可以對(duì)訓(xùn)練以及評(píng)估進(jìn)行可視化，下面有對(duì)應(yīng)的參數(shù)名稱：

loss，mean_absolute_error，val_loss，val_mean_absolute_error

import pandas as pd
import matplotlib.pyplot as plt
import argparse
import os
import numpy as np

def get_args():
 parser = argparse.ArgumentParser(description="This script shows training graph from history file.")
 parser.add_argument("--input", "-i", type=str, required=True,
      help="path to input history h5 file")
 args = parser.parse_args()
 return args

def main():
 args = get_args()
 input_path = args.input

 df = pd.read_hdf(input_path, "history")
 print(np.min(df['val_mean_absolute_error']))
 input_dir = os.path.dirname(input_path)
 plt.plot(df["loss"], '-o', label="loss (age)", linewidth=2.0)
 plt.plot(df["val_loss"], '-o', label="val_loss (age)", linewidth=2.0)
 plt.xlabel("Number of epochs", fontsize=20)
 plt.ylabel("Loss", fontsize=20)
 plt.legend()
 plt.grid()
 plt.savefig(os.path.join(input_dir, "loss.pdf"), bbox_inches='tight', pad_inches=0)
 plt.cla()

 plt.plot(df["mean_absolute_error"], '-o', label="training", linewidth=2.0)
 plt.plot(df["val_mean_absolute_error"], '-o', label="validation", linewidth=2.0)
 ax = plt.gca()
 ax.set_ylim([2,13])
 ax.set_aspect(0.6/ax.get_data_ratio())
 plt.xticks(fontsize=20)
 plt.yticks(fontsize=20)
 plt.xlabel("Number of epochs", fontsize=20)
 plt.ylabel("Mean absolute error", fontsize=20)
 plt.legend(fontsize=20)
 plt.grid()
 plt.savefig(os.path.join(input_dir, "performance.pdf"), bbox_inches='tight', pad_inches=0)

if __name__ == '__main__':
 main()

以上這篇在keras中實(shí)現(xiàn)查看其訓(xùn)練loss值就是小編分享給大家的全部?jī)?nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。

您可能感興趣的文章: