PyTorch Time-Series Loss Predictor

import torch
import torch.nn as nn

class ClaimsLSTM(nn.Module):
    def __init__(self):
        super().__init__()
        self.lstm = nn.LSTM(input_size=1, hidden_size=64, batch_first=True)
        self.fc = nn.Linear(64, 1)
        
# Train model on quarterly claims accumulation data...