code: def cric__decision_tree():
      
      model = sklearn.tree.DecisionTreeClassifier(random_state=0, max_depth=4)
  
      # we want to explain the raw probability outputs of the trees
      model.predict = lambda X: model.predict_proba(X)[:,1]
      
      return model
docs: Decision Tree