class_docstr: Simple uni-modal output model. 
code: def __init__(self, input_dim: int, output_dim: int):
      super().__init__()
      self.fc1 = nn.Linear(input_dim, output_dim, bias=False)
      self.to(self.device)
  
imports: from typing import List
  import numpy
  import pytest
  import torch
  import torch.nn
  import torch.optim
  from torch.utils.data import DataLoader, TensorDataset
  from alibi.models.pytorch.metrics import AccuracyMetric, LossContainer
  from alibi.models.pytorch.model import Model
label_desc: Use image node_id 1 to create a new UnimodalModel object from inherited base classes: Model with example: obj = UnimodalModel(input_dim, output_dim)