code: def build_linear_subclass_model(input_shape, weights, num_output=1):
      del input_shape
  
      class LinearModel(tf.keras.Model):
          def __init__(self):
              super(LinearModel, self).__init__()
              self.dense = tf.keras.layers.Dense(
                  num_output,
                  use_bias=False,
                  name="dense",
                  kernel_initializer=tf.keras.initializers.Constant(
                      weights
                  ),
              )
  
          def call(self, inputs):
              return self.dense(inputs[FEATURE_NAME])
  
      return LinearModel()
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  from absl.testing import parameterized
  from neural_structured_learning import configs
  from research.carls import graph_regularization
  from research.carls import neighbor_cache_client
  import numpy
  import tensorflow
  from google.protobuf import text_format
  from tensorflow.python.framework import test_util
label_desc: Use image node_id 4 for calling a global function with example usage: build_linear_subclass_model(input_shape, weights, num_output) and returns: LinearModel, self