Как указать линейную модель при построении в matplotlib - PullRequest
0 голосов
/ 16 мая 2019

У меня есть линейная модель в виде:

Результат ~ предсказатель_вариабельная + группа

Я хотел бы построить представление этого, включая доверительные интервалы.

IЯ могу правильно построить модель, но не могу визуализировать доверительные интервалы.Здесь я добавляю несколько линий регрессии, основанных на предсказанных значениях, которые я получаю из подобранной модели:

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 для моей желаемой модели.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...