Agregar barras de error a los valores predichos en Python

Sep 17 2020

Ejecuté un modelo lstm 10 veces y promedié los valores para obtener la media y ese se convirtió en mi valor predicho. Calculé la división estándar de cada fila y la agregué a mi marco de datos (test_iterations) que contiene mis valores predichos. Ahora quiero trazar barras de error para que mi gráfico se vea así (eso se hizo en Excel)

Pero cuando uso el siguiente código, obtengo un gráfico que se ve así

plt.errorbar(test_iterations['Mean'].values, 
                     test_iterations['Mean'].values,
                     yerr = test_iterations['Stdv'].values,
                     fmt='-o')

No estoy seguro de que lo que estoy haciendo sea correcto porque agregué valores xey en el código de la misma cosa, pero no estaba seguro de qué agregar allí.

Respuestas

2 GustaveCoste Sep 17 2020 at 07:45

El primer argumento de plt.errorbar()son las coordenadas x de los puntos a graficar. En su caso, es el rango del valor.

Prueba esto:

import numpy as np

plt.errorbar(np.arange(len(test_iterations['Mean'])),
             test_iterations['Mean'].values,
             yerr=test_iterations['Stdv'].values,
             fmt='-o')