Ejemplo n.º 1
0
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