I’m using an RNN with LSTM layers, and I am trying to make a prediction on a test set of 160 steps. Currently, after training my model, I call the MultilayerNetwork.output method on an INDArray of my test data’s features. When I do this, it gives a long array the size of my test data set. What method should I use if I want to output only the prediction on the last time step? So, instead of what I have now where it gives the prediction for the label on each step,1-160, I want it to make a prediction for the 161st step.
Related topics
| Topic | Replies | Views | Activity | |
|---|---|---|---|---|
| Getting output prediction for LSTM | 3 | 590 | September 24, 2022 | |
| Prediction of Disk Space | 8 | 455 | June 10, 2021 | |
| DL4J Need help with my input data | 5 | 499 | December 25, 2020 | |
| A post in "Weird results from my LSTM prediction" requires staff attention | 2 | 436 | January 11, 2021 | |
| Training data shapping | 4 | 738 | June 5, 2021 |