code: def __init__(self, **kwargs):
      super().__init__()
  
      self.train_acc = create_multiclass_accuracy()
      self.val_acc = create_multiclass_accuracy()
      self.test_acc = create_multiclass_accuracy()
      self.args = kwargs
  
      self.fc1 = nn.Linear(4, 10)
      self.fc2 = nn.Linear(10, 10)
      self.fc3 = nn.Linear(10, 3)
      self.cross_entropy_loss = nn.CrossEntropyLoss()
  
imports: import tempfile
  import pytorch_lightning
  import torch
  import torch.nn.functional
  import torch.utils.tensorboard
  from packaging.version import Version
  from torch import nn
label_desc: Use image node_id 1 to create a new IrisClassificationBase object from inherited base classes: pl with example: obj = IrisClassificationBase()