mirror of
https://github.com/saymrwulf/prophet.git
synced 2026-05-16 21:00:16 +00:00
f16e22731f
* Update and rename README to README.md and Unicode strings Updated Readme file and converted from rst to markdown. String contains ascii characters (converted to unicode string) Signed-off-by: Mpho Mphego mpho112@gmail.com * Deprecated import `from __future__ import unicode_literals` removed and ran isort module https://mail.python.org/pipermail/python-dev/2016-December/147009.html Included setuptool-git in the requirement.txt and updated `setup.py` Reasons for this are highlighted here -> https://github.com/msabramo/setuptools-git#usage
69 lines
2.1 KiB
Python
69 lines
2.1 KiB
Python
# -*- coding: utf-8 -*-
|
|
# Copyright (c) 2017-present, Facebook, Inc.
|
|
# All rights reserved.
|
|
#
|
|
# This source code is licensed under the BSD-style license found in the
|
|
# LICENSE file in the root directory of this source tree. An additional grant
|
|
# of patent rights can be found in the PATENTS file in the same directory.
|
|
|
|
from __future__ import absolute_import, division, print_function
|
|
|
|
import warnings
|
|
|
|
import numpy as np
|
|
import pandas as pd
|
|
|
|
import fbprophet.hdays as hdays_part2
|
|
import holidays as hdays_part1
|
|
|
|
|
|
def get_holiday_names(country):
|
|
"""Return all possible holiday names of given country
|
|
|
|
Parameters
|
|
----------
|
|
country: country name
|
|
|
|
Returns
|
|
-------
|
|
A set of all possible holiday names of given country
|
|
"""
|
|
years = np.arange(1995, 2045)
|
|
try:
|
|
with warnings.catch_warnings():
|
|
warnings.simplefilter("ignore")
|
|
holiday_names = getattr(hdays_part2, country)(years=years).values()
|
|
except AttributeError:
|
|
try:
|
|
holiday_names = getattr(hdays_part1, country)(years=years).values()
|
|
except AttributeError:
|
|
raise AttributeError(
|
|
"Holidays in {} are not currently supported!".format(country))
|
|
return set(holiday_names)
|
|
|
|
|
|
def make_holidays_df(year_list, country):
|
|
"""Make dataframe of holidays for given years and countries
|
|
|
|
Parameters
|
|
----------
|
|
year_list: a list of years
|
|
country: country name
|
|
|
|
Returns
|
|
-------
|
|
Dataframe with 'ds' and 'holiday', which can directly feed
|
|
to 'holidays' params in Prophet
|
|
"""
|
|
try:
|
|
holidays = getattr(hdays_part2, country)(years=year_list)
|
|
except AttributeError:
|
|
try:
|
|
holidays = getattr(hdays_part1, country)(years=year_list)
|
|
except AttributeError:
|
|
raise AttributeError(
|
|
"Holidays in {} are not currently supported!".format(country))
|
|
holidays_df = pd.DataFrame(list(holidays.items()), columns=['ds', 'holiday'])
|
|
holidays_df.reset_index(inplace=True, drop=True)
|
|
holidays_df['ds'] = pd.to_datetime(holidays_df['ds'])
|
|
return (holidays_df)
|