code: def fn(X):
      return model.predict(X)
  
imports: import shap
  import xgboost
  from sklearn.model_selection import train_test_split
  import mlflow
label_desc: Use image node_id 1 for calling a global function with example usage: fn(X) and returns: model