torch transformers lightgbm shap matplotlib sklearn ipython