lstmPend.m
ffd58092ca58master

lstmPend.m
View Options

			% lstm on data
			%% parameters
			printGraph = 0;%print the generated graph in eps
			fileNameInput = 'pendulum.csv';
			seq_length = 5;
			sq = int2str(seq_length);
			% parameters: rationTrain, seq_length, nbNodeHiddenLayer, epoch, batch size
			parameters = [' 0.6 ', sq,' 20 100 10'];
			command = ['python lstm.py ../input/', fileNameInput, parameters];
			system(command);% execute python script with the parameters
			%% load generated data
			load('../output/Otrain.mat')
			load('../output/Oval.mat')
			load('../output/score.mat')
			Ltrain = score(1);
			Lval = score(2);
			data = csvread(['../input/',fileNameInput]);
			t = data(:,1);
			q = data(:,2);
			%% Define useful variables
			l_t = length(Otrain);
			l_val = length(Oval);
			N = length(t);

			%% plot
			figure(1)
			set(gca,'FontSize',12)
			% set(gcf,'units','normalized','outerposition',[0 0 1 1])
			hold on
			plot(t,q);
			plot(t(seq_length + 1:l_t + seq_length),Otrain,'b')
			plot(t(l_t + 2*seq_length + 1:N),Oval,'r')
			xlabel('time [s]')
			ylabel('\theta [rad]')
			grid on
			if printGraph == 1
			figg = gcf;
			figg.PaperUnits = 'centimeters';
			figg.PaperPosition = [0 0 26 13];
			print('linPend','-depsc');
			end
			hold off