Extracting coefficients from GLM in Python using statsmodel
Asked Answered
F

1

17

I have a model which is defined as follows:

import statsmodels.formula.api as smf
model = smf.glm(formula="A ~ B + C + D", data=data, family=sm.families.Poisson()).fit()

The model has coefficients which look like so:

Intercept   0.319813
C[T.foo]   -1.058058
C[T.bar]   -0.749859
D[T.foo]    0.217136
D[T.bar]    0.404791
B           0.262614

I can grab the values of the Intercept and B by doing model.params.Intercept and model.params.B but I can't get the values of each C and D.

I have tried model.params.C[T.foo] for example, and I get and error.

How would I get particular values from the model?

Fingerbreadth answered 20/3, 2015 at 11:35 Comment(1)
A similar question: #47388758Welton
S
20

model.params is is a pandas.Series. Accessing as attribute is only possible if the name of the entry is a valid python name.

In this case you need to index with the name in quotes, i.e. model.params["C[T.foo]"]

see http://pandas.pydata.org/pandas-docs/dev/indexing.html

Standpipe answered 20/3, 2015 at 12:12 Comment(0)

© 2022 - 2024 — McMap. All rights reserved.