class_docstr: Models a simple one variable(y = wx) neural network, and a MSE loss function.
class_docstr_tok: ["Models","a","simple","one","variable","(","y","=","wx",")","neural","network",",","and","a","MSE","loss","function","."]
code: def build_model(self) -> Sequential:
      model = Sequential()
      model.add(
          Dense(
              1,
              activation=None,
              use_bias=False,
              kernel_initializer="zeros",
              input_shape=(1,),
          )
      )
      model = self.context.wrap_model(model)
  
      optimizer = SGD(learning_rate=self.my_lr)
      optimizer = self.context.wrap_optimizer(optimizer)
  
      model.compile(optimizer, mean_squared_error, metrics=["accuracy"])
  
      return model
  
file_path: determined/harness/tests/experiment/fixtures/ancient-checkpoints/0.17.7-keras/code/model_def.py
imports: import numpy
  import tensorflow
  from tensorflow.keras.layers import Dense
  from tensorflow.keras.losses import mean_squared_error
  from tensorflow.keras.models import Sequential
  from tensorflow.keras.optimizers.legacy import SGD
  from tensorflow.keras.utils import Sequence
  from determined.keras import TFKerasTrial
label_desc: Use image node_id 2 for calling the OneVarTrial obj's underlying member method code with example usage: obj.build_model() and returns: model