zhouwenyu0921 / python_stepwiseselection

Automated Backward and Forward Selection On Python

Automated Stepwise Backward and Forward Selection

This script is about an automated stepwise backward and forward feature selection. You can easily apply on Dataframes. Functions returns not only the final features but also elimination iterations, so you can track what exactly happend at the iterations.

You can apply it on both Linear and Logistic problems. Eliminations can be apply with Akaike information criterion (AIC), Bayesian information criterion (BIC), R-squared (Only works with linear), Adjusted R-squared (Only works with linear). Also you don't have to worry about varchar variables, code will handle it for you.

Enjoy the code!

Required Libraries: pandas, numpy, statmodels

See more about stepwise regression : https://en.wikipedia.org/wiki/Stepwise_regression