/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 vote295 views1 answer
How to add multiple items in a list in python
Python
13 votes276 views1 answer
Python read file to variable
Python
1 vote272 views1 answer
For loop django template count
Python
1 vote325 views1 answer
Using bs4 to obtain html element by id
Python
0 vote258 views2 answers
Render template in django
Python
3 votes282 views1 answer
What is cpython
Python
9 votes308 views3 answers
Read only the first line python
Python
1 vote263 views3 answers
How to sort values in python
Python
0 vote292 views2 answers
Openpyxl _cells_by_row
Python
1 vote314 views1 answer
User group template tag django
Python