import torch
from torch import nn


torch.manual_seed(11)
MODEL_PATH = "weather-score-state-dict.pth"


class WeatherScoreModel(nn.Module):
    def __init__(self):
        super().__init__()
        self.layers = nn.Sequential(
            nn.Linear(3, 5),
            nn.ReLU(),
            nn.Linear(5, 1),
        )

    def forward(self, inputs):
        return self.layers(inputs)