GaussianProcesses/data_presenter.py at master · gdroguski/GaussianProcesses · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
import os
import data_plotter

companies = []
plotters = {}
start_year = 2008


def main():
    __init_data()
    for company in companies:
        make_summary(company)
    print("Done!")


def make_summary(company_name):
    plotter = plotters[company_name]

    plotter.show_whole_time_series()
    plotter.show_time_series(start_year=start_year, end_year=2016)
    plotter.show_preprocessed_prices(start_year=start_year, end_year=2016)
    plotter.show_gp_prediction(train_start=start_year, train_end=2016, pred_year=2017)
    plotter.show_time_series(start_year=start_year, end_year=2018)
    plotter.show_gp_prediction(train_start=start_year, train_end=2018, pred_year=2018, pred_quarters=[3, 4])
    print(company_name + ' summary done!')


def __init_data():
    for company in os.listdir('Data'):
        current_company = company.split('.')[0]
        companies.append(current_company)
        plotters[current_company] = (data_plotter.Plotter(company_name=current_company))


if __name__ == "__main__":
    main()