input: def create_linear_regression_model(input_dim: int, output_dim: int) -> LinearRegressionModel:
      """ Instantiates a LinearRegressionModel class for a linear regression function.
      >>> model = create_linear_regression_model(5, 1)
      >>> model.weights.shape
      (5, 1)
      >>> model.bias.shape
      (1, 1)
      """
output:     class LinearRegressionModel:
          def __init__(self, input_dim, output_dim):
              self.weights = np.random.randn(input_dim, output_dim)
              self.bias = np.random.randn(1, output_dim)
  
      return LinearRegressionModel(input_dim, output_dim)