code: def forward(self, x):
      x = F.relu(self.fc1(x))
      x = F.relu(self.fc2(x))
      x = F.dropout(x, 0.2)
      x = self.fc3(x)
      return x
  
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 2 for calling the IrisClassifier obj's underlying member method code with example usage: obj.forward(x) and returns: x