модель смешанного эффекта с питоном (statemodels) - PullRequest
Новая
       70

модель смешанного эффекта с питоном (statemodels)

0 голосов
/ 17 марта 2019

У меня есть набор данных с этими столбцами:

«sampno_vehno», «vehicle_age», «vehicle_body_type», «vehicle_fuel_type», «trip», «rushHours», «Sunday», «duration_min» и «volatility».

Я хочу использовать модель смешанных эффектов с mixedlm в satemodels (библиотека python), но я не знаю, как ее кодировать.

Я попробовал следующий код, и я знаю, что у него есть какая-то проблема. 

import statsmodels.formula.api as smf
import pandas as pd

df = pd.DataFrame(myData, columns=['sampno_vehno', 'vehicle_age', 'vehicle_body_type',
                                   'vehicle_fuel_type', 'trip', 'rushHours', 'weekend',
                                   'duration_min', 'volatility'], dtype=float)

model = 'volatility ~ vehicle_age + C(vehicle_body_type) + C(vehicle_fuel_type) + C(rushHours) + C(weekend) + duration_min + trip'
md = smf.mixedlm(model, df, groups=df['sampno_vehno'],re_formula = "~trip")
mdf = md.fit()
print(mdf.summary())
...