code: def __init__(self, n_features: int):
      super().__init__()
      self.dense1 = nn.Linear(n_features, 20)
      self.dense2 = nn.Linear(20, 2)
  
file_path: alibi-detect/alibi_detect/cd/keops/tests/test_mmd_keops.py
imports: from functools import partial
  from itertools import product
  import numpy
  import pytest
  import torch
  import torch.nn
  from typing import Callable, List
  from alibi_detect.utils.frameworks import has_keops
  from alibi_detect.utils.pytorch import GaussianRBF, mmd2_from_kernel_matrix
  from alibi_detect.cd.pytorch.preprocess import HiddenOutput, preprocess_drift
label_desc: Use image node_id 1 to create a new MyModel object from inherited base classes: nn with example: obj = MyModel(n_features)