print(modelo.conf_int(alpha=0.05)) print('') # Predicciones: # Una vez entrenado el modelo, se pueden obtener predicciones # para nuevos datos. Los modelos de statsmodels permiten # calcular las predicciones de dos formas: # .predict(): devuelve únicamente el valor de las predicciones. # .get_prediction().summary_frame(): devuelve, además de # las predicciones, los intervalos de confianza asociados. predicciones = modelo.get_prediction(exog = X_train).summary_frame(alpha=0.05) predicciones.head(4) print('data - predicciones') print(predicciones) # Predicciones con intervalo de confianza del 95% # ============================================================================== predicciones = modelo.get_prediction(exog = X_train).summary_frame(alpha=0.05) predicciones['x'] = X_train[:, 1] predicciones['y'] = y_train predicciones = predicciones.sort_values('x') # Gráfico del modelo