天天看點

keras訓練過程可視化

http://blog.csdn.net/xiaojiajia007/article/details/72865764

https://stackoverflow.com/questions/42112260/how-do-i-use-the-tensorboard-callback-of-keras

https://www.tensorflow.org/get_started/summaries_and_tensorboard

直接上代碼

reduce_lr_loss = callbacks.ReduceLROnPlateau(monitor='val_loss', factor=0.2, patience=5, min_lr=0.001)

tb_cb=keras.callbacks.TensorBoard(log_dir='./logs', histogram_freq=1, write_graph=True, write_images=False, embeddings_freq=0, embeddings_layer_names=None, embeddings_metadata=None)

es_cb=EarlyStopping(monitor='val_loss', min_delta=0.09,patience=5, verbose=0, mode='auto')

cbks=[]

cbks.append(tb_cb)

cbks.append(es_cb)

cbks.append(reduce_lr_loss)

model.fit(x_train, y_train,batch_size=batch_size,epochs=epochs,verbose=1,callbacks=cbks,validation_data=(x_test, y_test))

需要檢視的時候,在指令行視窗  tensorboard --logdir=C:\Users\Alexander\logs(這裡是log_dir的位置)

然後在浏覽器輸入  http://localhost:6006,可以檢視

history =

model.fit(x_train, y_train,batch_size=batch_size,epochs=epochs,verbose=1,callbacks=cbks,validation_data=(x_test, y_test))

fig = plt.figure()

plt.plot(history.history['acc'])

plt.plot(history.history['val_acc'])

plt.title('model accuracy')

plt.ylabel('accuracy')

plt.xlabel('epoch')

plt.legend(['train', 'test'], loc='upper left')

plt.plot(history.history['loss'])

plt.plot(history.history['val_loss'])

plt.title('model loss')

plt.ylabel('loss')

plt.xlabel('epoch')

plt.legend(['train', 'test'], loc='lower left')

fig.savefig('performance.png')

keras訓練過程可視化

繼續閱讀