/Python/Feature importance naive bayes python

Feature importance naive bayes python

neg_class_prob_sorted = NB_optimal.feature_log_prob_[0, :].argsort()[::-1]
pos_class_prob_sorted = NB_optimal.feature_log_prob_[1, :].argsort()[::-1]

print(np.take(count_vect.get_feature_names(), neg_class_prob_sorted[:10]))
print(np.take(count_vect.get_feature_names(), pos_class_prob_sorted[:10]))

Source: stackoverflow.com

Similar codes

1 vote171 views1 answer
Flask set cookie
Python
0 vote166 views1 answer
Python dt error only use with datetimelike values
Python
0 vote185 views1 answer
Supercharged python
Python
-1 vote159 views1 answer
Numpy apply function to array
Python
0 vote216 views1 answer
Integer to boolean numpy
Python
6 votes161 views3 answers
How to remove all zeros from a list in python
Python
1 vote181 views1 answer
Python add one
Python
0 vote176 views1 answer
Add variable in text python
Python
0 vote162 views1 answer
Pytorch cuda tensor in module
Python
1 vote160 views2 answers
Speech enhancement techniques
Python