docstr: Args:
      train_data
          Train data to use
      kwargs
          Arguments to feed to `.fit` method for sklearn
code: def fit(self, train_data: DataLoader, **kwargs):
      r"""
      Args:
          train_data
              Train data to use
          kwargs
              Arguments to feed to `.fit` method for sklearn
      """
      return super().fit(
          train_data=train_data,
          sklearn_trainer=self.sklearn_module,
          **kwargs,
      )
  
imports: from typing import Callable, cast, List, Optional
  import torch.nn
  from captum._utils.models.model import Model
  from torch import Tensor
  from torch.utils.data import DataLoader
label_desc: Use image node_id 2 for calling the SkLearnLinearModel obj's underlying member method code with example usage: obj.fit(train_data) and returns: super