/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

0 vote184 views1 answer
Exercises with classes in python with solutions
Python
1 vote168 views1 answer
Matplotlib remove x and y labels
Python
77 votes187 views6 answers
How to sort a dictionary by value in python
Python
3 votes188 views1 answer
Pandas loc for list
Python
0 vote207 views1 answer
Show all of a truncated dataframe jupyter"
Python
4 votes192 views1 answer
Server on python with address and port
Python
1 vote221 views1 answer
Python convert file into list
Python
0 vote216 views1 answer
Python declare variable type array
Python
3 votes225 views1 answer
From collections import defaultdict
Python
1 vote155 views1 answer
Prime number between range program in python
Python