code: def __init__(self):
      super().__init__()
      self.fc1 = nn.Linear(1, 50)
      self.relu1 = nn.ReLU()
      self.dout = nn.Dropout(0.2)
      self.fc2 = nn.Linear(50, 100)
      self.prelu = nn.PReLU(1)
      self.out = nn.Linear(100, 1)
      self.out_act = nn.Sigmoid()
  
imports: import pytest
  import logging
  from unittest import TestCase
  import ray
  from ray.data import Dataset
  import torch.nn
  import torch.optim
  from bigdl.orca import init_orca_context, stop_orca_context
  from bigdl.orca.learn.pytorch import Estimator
  from bigdl.orca.learn.metrics import MSE
label_desc: Use image node_id 1 to create a new Net object from inherited base classes: nn with example: obj = Net()