code: def __init__(self):
      super(LogisticRegression, self).__init__()
      self.linear1 = nn.Linear(2, 4)
      self.linear2 = nn.Linear(4, 1)
  
file_path: pytorch-optimizer/tests/test_optimizer_with_nn.py
imports: import numpy
  import pytest
  import torch
  from torch import nn
  import torch_optimizer
label_desc: Use image node_id 1 to create a new LogisticRegression object from inherited base classes: nn with example: obj = LogisticRegression()