One of the primary tasks involved in any supervised Machine Learning venture is to select the best features from the given dataset to obtain the best results. One way to select these features is the Chi-Square Test.
Mathematically, a Chi-Square test is done on two distributions two determine the level of similarity of their respective variances. In its null hypothesis, it assumes that the given distributions are independent. This test thus can be used to determine the best features for a given dataset by determining the features on which the output class label is most dependent on. For each feature in the dataset, the is calculated and then ordered in descending order according to the value. The higher the value of , the more dependent the output label is on the feature and higher the importance the feature has on determining the output.
Let the feature in question have m attribute values and the output have k class labels. Then the value of is given by the following expression:-
– Observed frequency
– Expected frequency
For each feature, a contingency table is created with m rows and k columns. Each cell (i,j) denotes the number of rows having attribute feature as i and class label as k. Thus each cell in this table denotes the observed frequency. To calculate the expected frequency for each cell, first the proportion of the feature value in the total dataset is calculated and then it is multiplied by the total number of the current class label.
Consider the following table:-
Here the output variable is the column named “PlayTennis” which determines whether tennis was played on the given day given the weather conditions.
The contingency table for the feature “Outlook” is constructed as below:-
Note that the expected value for each cell is given inside the parenthesis.
The expected value for the cell (Sunny,Yes) is calculated as and similarly for others.
The value is calculated as below:-
The contingency table for the feature “Wind” is constructed as below:-
The value is calculated as below:-
On comparing the two scores, we can conclude that the feature “Wind” is more important to determine the output than the feature “Outlook”.
This article demonstrates how to do feature selection using Chi-Square Test.
- ML | Chi-square Test for feature selection
- Parameters for Feature Selection
- ML | Mathematical explanation of RMSE and R-squared error
- Mathematical explanation for Linear Regression working
- ML | Extra Tree Classifier for Feature Selection
- numpy.random.chisquare() in Python
- ML | Feature Mapping
- Sklearn | Feature Extraction with TF-IDF
- Feature Extraction Techniques - NLP
- ML | Feature Scaling – Part 2
- ML | Feature Scaling - Part 1
- Python | How and where to apply Feature Scaling?
- PyQt5 QDockWidget – Getting Feature change signal
- Feature Encoding Techniques - Machine Learning
- Feature matching using ORB algorithm in Python-OpenCV
- Q-learning Mathematical Background
- ML | OPTICS Clustering Explanation
- Explanation of BERT Model - NLP
- Python | Mathematical Median of Cumulative Records
- Plot Mathematical Expressions in Python using Matplotlib
If you like GeeksforGeeks and would like to contribute, you can also write an article using contribute.geeksforgeeks.org or mail your article to email@example.com. See your article appearing on the GeeksforGeeks main page and help other Geeks.
Please Improve this article if you find anything incorrect by clicking on the "Improve Article" button below.
Improved By : ManasChhabra2