Skip to content

Instantly share code, notes, and snippets.

@ortsed

ortsed/gnb_importance.py

Created November 11, 2020 18:00
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save ortsed/2389ba88913670b218367fe773ab68f8 to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save ortsed/2389ba88913670b218367fe773ab68f8 to your computer and use it in GitHub Desktop.
Download ZIP
Gaussian Naive Bayes Importance
Raw
gnb_importance.py
import numpy as np
neg = model.theta_[0].argsort()
print(np.take(X_test.columns, neg[:10]))
print('')
neg = model.sigma_[0].argsort()
print(np.take(X_test.columns, neg[:10]))
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment