Skip to content

Instantly share code, notes, and snippets.

@stasSajin

stasSajin/extract_interactions.py

Created October 31, 2018 03:52
Show Gist options
  • Save stasSajin/f751f3047babbf1949f43ec6b3763b0a to your computer and use it in GitHub Desktop.
Save stasSajin/f751f3047babbf1949f43ec6b3763b0a to your computer and use it in GitHub Desktop.
Download ZIP
Raw
extract_interactions.py
feature_names = dtest.feature_names
interactions = xgb_model.predict(dtest, pred_interactions=True)
pd.DataFrame(interactions[0],
index = feature_names + ['intercept'],
columns= feature_names + ['intercept'])
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment