How can I extract metrics/data of the LSTM training progress ?

11 vues (au cours des 30 derniers jours)
Pierre Guy
Pierre Guy le 10 Oct 2019
Commenté : Pierre Guy le 16 Oct 2019
While monitoring a deep learning (LSTM) training progress, how can I extract the quantities data (Training root mean square error (RMSE) , Smoothed training accuracy, Validation accuracy, Training loss, smoothed training loss, and validation loss ) at each iteration ?( suppose you want to export these data to excel).
Indeed, while training a deep (LSTM) network in matab, '' when you specify 'training-progress' as the 'Plots' value in trainingOptions and start network training, trainNetwork creates a figure and displays training metrics at every iteration. Each iteration is an estimation of the gradient and an update of the network parameters. If you specify validation data in trainingOptions, then the figure shows validation metrics each time trainNetwork validates the network. The figure plots the following: Training root mean square error (RMSE) , Smoothed training accuracy, Validation accuracy, Training loss, smoothed training loss, and validation loss."
Thank you.
Best regards,

Réponse acceptée

Jalaj Gambhir
Jalaj Gambhir le 14 Oct 2019
Hi,
You can use
[net,info] = trainNetwork(XTrain,YTrain,layers,options);
while training a network, to access 'info' struct. This struct contains information such as 'TrainingLoss', 'TrainingRMSE' and 'BaseLearnRate' for each iteration of training. You can refer to the documentation for more information.

Plus de réponses (0)

Produits


Version

R2018a

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!

Translated by