После запуска формы кода пример, найденный в документации класса GLMgam пакета statsmodel,
from statsmodels.gam.api import GLMGam, BSplines
from statsmodels.gam.tests.test_penalized import df_autos
df_autos_train = df_autos.iloc[0:165]
df_autos_test = df_autos.iloc[165:205]
x_spline = df_autos_train[['weight', 'hp']]
bs = BSplines(x_spline, df=[12, 10], degree=[3, 3])
alpha = np.array([21833888.8, 6460.38479])
gam_bs = GLMGam.from_formula('city_mpg ~ fuel + drive', data=df_autos_train, smoother=bs, alpha=alpha)
res_bs = gam_bs.fit()
print(res_bs.summary())
Мне не удается получить прогноз
res_bs.predict(exog = df_autos_test[['fuel', 'drive']], exog_smooth = df_autos_test[['weight', 'hp']])
дает мне ошибка
File "pandas\_libs\index.pyx", line 111, in pandas._libs.index.IndexEngine.get_loc
File "pandas\_libs\index.pyx", line 116, in pandas._libs.index.IndexEngine.get_loc
TypeError: '(slice(None, None, None), 0)' is an invalid key