"""StatsModels Example"""

import numpy as np
import scikits.statsmodels.api as sm

# MULTIPLE REGRESSION
# get data
nsample = 100
x = np.linspace(0,10, 100)
X = sm.add_constant(np.column_stack((x, x**2)))
beta = np.array([1, 0.1, 10])
y = np.dot(X, beta) + np.random.normal(size=nsample)

# run the regression
results = sm.OLS(y, X).fit()

# look at the results
print results.summary()


#GENERALIZED LINEAR MODEL
data = sm.datasets.scotland.load()
data.exog = sm.add_constant(data.exog)
poisson_model = sm.GLM(data.endog, data.exog,
                     family=sm.families.Poisson())
poisson_results = poisson_model.fit()
print "Fitted Parameters:"
print poisson_results.params
print
print "P-values"
print poisson_results.pvalues