Views
No views yet
1import torch
2from model import TradingLSTM
3
4model = TradingLSTM()
5model.load_state_dict(torch.load('trading_lstm.pth'))
6model.eval()
7
8# Prepare input sequence (50, 16)
9# Predict probability of up move
10prediction = model(sequence)