Постройте ошибки на линейном графике - PullRequest
Новая
       7

Постройте ошибки на линейном графике

0 голосов
/ 04 мая 2020

Вот мой текущий график:

enter image description here

Я бы хотел добавить полосы ошибок на оранжевую линию, где я проводил проверочный тест. ..

Так выглядят необработанные переменные 

train : list of tuples where the first position is batch number and second is the 
        associated loss
        e.g. [(batch, loss), (batch, loss), ...]
test  : list of tuples where the first position is batches processed when the  
        validation was conducted, and the second position is a list of batch losses
        e.g. [(batch, [loss, loss, ...]), (batch, [loss, loss, ...]), ...]

Вот как я построил график 

plt.style.use('ggplot')
df_train = pd.DataFrame(train, columns=["Batch","Train"])
df_val   = pd.DataFrame([(batch,sum(losses)/len(losses)) for batch, losses in test], # avg losses from test cycle
                         columns=["Batch","Test"])
data  = df_train.merge(df_val, "outer").sort_values("Batch")
data.interpolate(method="linear").plot(x="Batch", color=["#2976bf","#d57f2e"])

Должен ли я как-то наложить график plt.errorbar на вот этот? Каков был бы правильный подход к этой проблеме? Я счастлив использовать другую библиотеку для построения графиков, если для этого есть лучший инструмент.

...