import ai.djl.ndarray.NDList;
import ai.djl.ndarray.types.Shape;
import ai.djl.nn.recurrent.LSTM;
import ai.djl.training.ParameterStore;
import java.util.List;
public class Predict {
public static List<Main.Candle> predict(List<Main.Candle> candles, int nPredicts) {
LSTM lstm = new LSTM.Builder().setNumLayers(4).build();
lstm.prepare(new Shape[]{new Shape(candles.size() - 1, 4)});
NDList forward = lstm.forward(new ParameterStore(), new NDList(candles.size() - 2), true);
//???
for (int i = 0; i < candles.size() - 1; i++) {
Main.Candle c = candles.get(i);
System.out.println(c.openTime() + " " + c.closeTime() + " " + c.open() + " " + c.low() + " " + c.high() + " " + c.close());
}
}
}