import torch
from torch import nn


torch.manual_seed(7)


class SensorRegressor(nn.Module):
    def __init__(self, in_features=4, hidden_features=8, out_features=2):
        super().__init__()
        self.input = nn.Linear(in_features, hidden_features)
        self.activation = nn.ReLU()
        self.output = nn.Linear(hidden_features, out_features)