如何使用Patsy创建所有可能的公式组合以进行模型选择?

时间:2017-03-08 00:17:19

标签: python statsmodels patsy

我目前正在使用Python的Patsy模块为我的模型创建矩阵输入。例如,我可能使用的公式是

'Survived ~ C(Pclass) + C(Sex) + C(honor) + C(tix) + Age + SibSp + ParCh + Fare + Embarked + vowel + middle + C(Title)'

但是,我想进行模型选择,所以我想从最简单的模型中创建所有可能的公式

'Survived ~ Age'

到最复杂的

模型
'Survived ~ C(Pclass) * C(Sex) * C(honor) * C(tix) * Age * SibSp * ParCh * Fare * Embarked * vowel * middle * C(Title)'

Patsy中是否有命令或某种方式我可以生成所有可能的字符串组合?

1 个答案:

答案 0 :(得分:1)

这似乎是一个简单的字符串生成问题:

import itertools

survived = 'Survived'
operators = '+ - * / : **'.split()
factors = """C(Pclass) C(Sex) C(honor) C(tix) Age SibSp ParCh Fare Embarked vowel middle C(Title)""".split()

for l in range(1,len(factors)):
    for fax in itertools.permutations(factors, l):
        for ops in itertools.product(operators, repeat=(l-1)):
            expr = [val for pair in itertools.zip_longest(fax, ops) for val in pair if val is not None]
            print(survived, '~', ' '.join(expr))