在matplotlib中进行绘图时如何指定线性模型

时间:2019-05-16 10:50:53

标签: python matplotlib plot linear-regression seaborn

我有以下形式的线性模型:

结果〜预报变量+组

我想绘制一个表示形式,包括置信区间。

我能够正确绘制模型,但无法可视化置信区间。在这里,我根据拟合模型得出的预测值添加了一些回归线:

fig, ax = plt.subplots() 
sns.scatterplot(x='predictor', y='true_outcome', hue = 'group', data=df, ax=ax)
sns.regplot(x='predictor,y='predicted_outcome',scatter=False, data = 
            df[df['group']==0], ax=ax)
sns.regplot(x='predictor,y='predicted_outcome',scatter=False, data = 
            df[df['group']==1], ax=ax)

enter image description here

或者我可以可视化置信区间,但是该图不能准确地表示模型,因为这两条线是独立拟合的:

sns.lmplot(x='predictor', y='true_outcome',  hue='group', data = df)

enter image description here

我想要一些有关例如在第一张图上显示适当的置信区间,或如何获取lmplot以适合我所需的模型。

0 个答案:

没有答案