当我想在python中使用某个模型时,
我经常在fit()
中使用statsmodels
方法。
在某些情况下,我编写了一个自动拟合的脚本:
import statsmodels.formula.api as smf
import pandas as pd
df = pd.read_csv('mydata.csv') # contains column x and y
fitted = smf.poisson('y ~ x', df).fit()
我的问题是如何使fit()
方法沉默。
在我的环境中,它输出一些关于适合标准输出的信息,如:
Optimization terminated successfully.
Current function value: 2.397867
Iterations 11
但我不需要它。
我找不到控制标准输出打印的参数。
我如何沉默fit()
方法?
Python 3.3.4,IPython 2.0.0,pandas 0.13.1,statsmodels 0.5.0。
答案 0 :(得分:18)
使用disp参数来拟合。它控制着scipy中优化器的详细程度。
mod.fit(disp=0)
请参阅文档以了解适合。