code: def test__train_lm(self):
      lm = linear_model.SGDClassifier(alpha=self.hparams["alpha"])
      acc = self.cav._train_lm(
          lm,
          np.array([[0], [0], [0], [1], [1], [1]]),
          np.array([0, 0, 0, 1, 1, 1]),
          {0: 0, 1: 1},
      )
      # the given data is so easy it should get this almost perfect.
      self.assertGreater(acc[0], 0.99)
      self.assertGreater(acc[1], 0.99)
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import os
  import pickle
  import shutil
  import numpy
  import tensorflow
  from sklearn import linear_model
  from six.moves import range
  from tcav.cav import CAV, get_or_train_cav
  from tensorflow.python.platform import flags
  from tensorflow.python.platform import googletest
label_desc: Use image node_id 11 for calling the CavTest obj's underlying member method code with example usage: obj.test__train_lm() without return types