I want to store the data of MSE vs epoch obtained during training of the neural network. By using "plotperform(TR)" we are able to get the MSE vs epoch graph but not the data itself. How to store this MSE vs epoch data in array variable?

27 views (last 30 days)
how to obtain MSE vs Epoch data of training the neural network in MATLAB? %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
clc;clear all;close all;
net_size_array={[5] [10] [20] [5 5] [10 10] [20 20] [5 5 5] [10 10 10] [20 20 20]};
for i=1:size_training
[net, TR]=train(net,in,out);
plotperform_pk(TR);grid on;hold on;
%% I want to store and plot the MSE vs Epoch graph for various neural network size.

Answers (1)

Sai Praneet
Sai Praneet on 24 Oct 2020
Edited: Sai Praneet on 24 Oct 2020
In your code the line:
[net, TR]=train(net,in,out);
stores all the essential information in the structure variable "TR". To get the Epoch vs. MSE data corresponding to the training, validation, and testing set all we have to do is store them in some variables for each network size such as
train_mse = TR.perf ;
val_mse = TR.vperf;
test_mse = TR.tperf;

Community Treasure Hunt

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

Start Hunting!

Translated by