def shap_plot(my_model, val_X, class_num=1):
    """Create object that can calculate shap values"""
    import shap
    explainer = shap.TreeExplainer(my_model)
    shap_values = explainer.shap_values(val_X)
    if class_num is not None:
        shap_values = shap_values[class_num]
    shap.summary_plot(shap_values, val_X)
