How to calculate information gain for each attribute with respect to a class in a document term matrix using sklearn in Python?
By using mutual_info_classif from sklearn in python we can implement information gain. Below is the code from sklearn.datasets import fetch_20newsgroups from sklearn.feature_selection import mutual_info_classif from sklearn.feature_extraction.text import CountVectorizer categories = ['talk.religion.misc',…
Answered Nov 7, 2019 · 1.6K Views Read answer →