A package to make forecast visualisation easier

Objective: To make forecast visualisation easier and more expressive.

To install the package

pip install forecast-tool

Let me show you how the package works

(1) Make a simple forecast using ARIMA model

Input [1]:

from forecast_tool import forecast_plot as fp
df = fp.get_data()

period = 'y' #[y, m, d] 
date='Date'
newtarget='Volume'

fp.overall_vis(df, date, newtarget, period)

Output [1]:

(1) You can also customise it to create your own plot

Input [2]:

show_model_name=True
suggested_n_forcast=True
my_title = True
y_label_name = True
x_label_name = True
color_past='dodgerblue'
color_forecast='crimson'
linestyle_past = 'solid'
linestyle_forecast='--'
show_confidence=True
numElems=10
matplotlib_style = 'seaborn'
legend_position=2
fig_length=13.5
fig_height=6.5
legend_font_size=14
scientific_annotation=True
zoom_last_n = 3  # to zoom last n values in a plot

fp.overall_vis(df, date, newtarget, period, \
               zoom_last_n=zoom_last_n, matplotlib_style=matplotlib_style, scientific_annotation=False)

Output [2]:

This is contributed by Morris Lee.