/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 vote98 views1 answer
Run julia in p;ython
Python
33 votes98 views2 answers
Python b to string
Python
0 vote95 views1 answer
Dataframe, groupby, select one
Python
0 vote100 views1 answer
Python plot two lines with different y axis
Python
0 vote99 views1 answer
Django.db.utils.ProgrammingError: relation "users" does not exist in django 3.0
Python
0 vote63 views1 answer
How to concatenate two columns matrix in python
Python
0 vote97 views1 answer
Python read column data from text file
Python
0 vote97 views1 answer
Format numbers in dataframe pandas
Python
0 vote98 views1 answer
Matplotlib radial averaging
Python
0 vote102 views1 answer
Today's day of the year
Python