class_docstr: Test functionality of SkorchDoctor using a simple model
class_docstr_tok: ["Test","functionality","of","SkorchDoctor","using","a","simple","model"]
docstr: Return a simple module class with predictable parameters
docstr_tok: ["Return","a","simple","module","class","with","predictable","parameters"]
code: def module_cls(self):
      """Return a simple module class with predictable parameters"""
  
      class MyModule(nn.Module):
          """Module with predictable parameters"""
  
          def __init__(self):
              super().__init__()
  
              self.lin0 = nn.Linear(20, 20)
              nn.init.eye_(self.lin0.weight)
              nn.init.zeros_(self.lin0.bias)
  
              self.lin1 = nn.Linear(20, 2)
              nn.init.zeros_(self.lin1.weight)
              nn.init.ones_(self.lin1.bias)
  
              self.softmax = nn.Softmax(dim=-1)
  
          def forward(self, X):
              X = self.lin0(X)
              X = self.lin1(X)
              return self.softmax(X)
  
      return MyModule
  
imports: import itertools
  from unittest import mock
  import numpy
  import pytest
  import torch
  from torch import nn
label_desc: Use image node_id 1 for calling the TestSkorchDoctorSimple obj's underlying member method code with example usage: obj.module_cls() and returns: MyModule, self