Does 'statsmodels' or another Python package offer an equivalent to R's 'step' function?

orome Published at Dev

orome

Is there a statsmodels or other Python equivalent for R's step functionality for selecting a formula-based model using AIC?

Kostya

I really suspect that you are doing the same online course as I do -- the following allows you to get the right answers. If the task at hand is not very computationally heavy (and it isn't in the course), then we can sidestep all the smart details of the step function, and just try all the subsets of the predictors.

For each subset we can calculate AIC as ACI = 2*nvars - 2*result.llf.
And then we just find a subset with the minimal AIC:

import itertools
import numpy as np
import pandas as pd
import statsmodels.api as sm
AICs = {}
for k in range(1,len(predictorcols)+1):
    for variables in itertools.combinations(predictorcols, k):
        predictors = train[list(variables)]
        predictors['Intercept'] = 1
        res = sm.OLS(target, predictors).fit()
        AICs[variables] = 2*(k+1) - 2*res.llf
pd.Series(AICs).idxmin()

Collected from the Internet

Please contact [email protected] to delete if infringement.

edited at2021-02-7

Comments

0 comments

From Java

Related Related

Article

Does 'statsmodels' or another Python package offer an equivalent to R's 'step' function?

Does 'statsmodels' or another Python package offer an equivalent to R's 'step' function?

Equivalent of R's factor function in Pandas

equivalent of R's View for Python's pandas

Python equivalent of R's head and tail function

What is python's equivalent of R's NA?

Python equivalent of R "split"-function

Is there a Python equivalent to MATLAB's pearsrnd function?

Python's equivalent for R's dput() function

Python equivalent for R's 'zoo' package

Python numpy or pandas equivalent of the R function sweep()

Is there an equivalent to Python's all function in JavaScript or jQuery?

Python equivalent of daisy() in the cluster package of R

Equivalent of R's createDataPartition in Python

What is the Java equivalent to Python's reduce function?

R equivalent to the Python function "dir"?

Does Python have an equivalent of Perl's qq?

Equivalent of Python's dir function in PHP

Matlab equivalent of Python's 'reduce' function

Does Rust have an equivalent to Python's unichr() function?

Equivalent of R's removeSparseTerms in Python

Is there a Python equivalent to R's sample() function?

Is there an equivalent to Python's exec() function in Java?

Pandas equivalent of Python's readlines function

Is there a python ReportLab equivalent of TCPDF's 'annotate' function?

Does python have an equivalent to Javascript's 'btoa'

R: Is there an equivalent of Stata's ibn. function?

Is there a C# equivalent to python's type() function?

Python equivalent to Matlab's set function

What is the Python Equivalent of Batch's PAUSE function

Matlab equivalent of Python's 'reduce' function