code: def get_model(model_fn, train_data, param):
      
      model_param = merge_dicts({"train_data": train_data}, param["model"], param.get("shared", {}))
      return model_fn(**model_param)
docs: Feed model_fn with train_data and param