code: def __init__(self):
      super().__init__()
      self.fc1 = nn.Linear(4, 10)
      self.fc2 = nn.Linear(10, 10)
      self.fc3 = nn.Linear(10, 3)
  
file_path: mlflow/examples/pytorch/torchscript/IrisClassification/iris_classification.py
imports: import argparse
  import torch
  import torch.nn.functional
  from sklearn.datasets import load_iris
  from sklearn.metrics import accuracy_score
  from sklearn.model_selection import train_test_split
  from torch import nn
  import mlflow.pytorch
  from mlflow.models import infer_signature
label_desc: Use image node_id 1 to create a new IrisClassifier object from inherited base classes: nn with example: obj = IrisClassifier()