code: def test_pytorch_lr_classifier():
      """Assert that the PytorchLRClassifier model works. (NN)"""
      clf = nn.PytorchLRClassifier(num_epochs=1, batch_size=8)
      pred = clf.fit_transform(pd_features, pd_target)
      tags = clf._more_tags()
  
file_path: tpot/tests/nn_tests.py
imports: from tpot import TPOTClassifier
  from tpot.config import classifier_config_nn
  from tpot.builtins import nn
  import numpy
  import pandas
  from sklearn.datasets import make_classification
  from sklearn.model_selection import train_test_split
  from nose.tools import nottest, assert_raises
  from itertools import repeat
label_desc: Use image node_id 3 for calling a global function with example usage: test_pytorch_lr_classifier() without return types