U vb @ s8 d dl mZmZ d dlZd dlmZ dd Zdd Z dS ) )displayHTMLNc C s> |d }|d }|D ]}ttd| d |D ]}||d |k|d |k@ }tj|d }| | d |k }tjdd} t } | d| d | d | j tj|d |d dd dd | j||d |d ddd | j ||d ddd | j ||d ddd | d | d | t q:qd S )Nmodel customer_type
timestamp figsize Forecast z demand ()demand-blackActualfmtcolorlabelp10p90333333?z80% Confidence Intervalalphar p50zPrediction Medianr r meanzPrediction MeanDateRides Taken (Demand))uniquer r mpldatesdate2numpltfiguregca set_title plot_datefill_between set_xlabel set_ylabellegendshow)actualspredictionsmodelscustomer_typesr r predictionprediction_tsnumsactualfigax r9 H/home/ec2-user/SageMaker/time-series-forecasting-on-aws/util/plotting.pyplot_fcst_results s^ r; c C sH |d }|d }|D ]$}ttd| d | | d |k }tjdd}t }|d| |jtj |d |d d ddd |D ]}||d |k|d |k@ } tj | d } |j| | d | d dd| d |j| | d d d| d |j| | d d d| d q|d | d | t qd S )Nr r r r r r r r r r r r r r r r z80% Confidence Interval for r r zPrediction Median for r r zPrediction Mean for r r! )r" r r r&