mirror of
https://github.com/saymrwulf/prophet.git
synced 2026-05-14 20:48:08 +00:00
79d0793ce4
* Resolve conflict * Change comments and add error column to output DataFrame * Change file structure * Update * Modified diagnostics * Update diagnostics.py following the advice on Github * Add tests and documentation * Change copy method into Prophet class and reflect comments
125 lines
3.9 KiB
Python
125 lines
3.9 KiB
Python
import os.path
|
|
import pickle
|
|
import platform
|
|
import sys
|
|
|
|
from pkg_resources import (
|
|
normalize_path,
|
|
working_set,
|
|
add_activation_listener,
|
|
require,
|
|
)
|
|
from setuptools import setup
|
|
from setuptools.command.build_py import build_py
|
|
from setuptools.command.develop import develop
|
|
from setuptools.command.test import test as test_command
|
|
|
|
|
|
PLATFORM = 'unix'
|
|
if platform.platform().startswith('Win'):
|
|
PLATFORM = 'win'
|
|
|
|
SETUP_DIR = os.path.dirname(os.path.abspath(__file__))
|
|
MODELS_DIR = os.path.join(SETUP_DIR, 'stan', PLATFORM)
|
|
MODELS_TARGET_DIR = os.path.join('fbprophet', 'stan_models')
|
|
|
|
|
|
def build_stan_models(target_dir, models_dir=MODELS_DIR):
|
|
from pystan import StanModel
|
|
for model_type in ['linear', 'logistic']:
|
|
model_name = 'prophet_{}_growth.stan'.format(model_type)
|
|
target_name = '{}_growth.pkl'.format(model_type)
|
|
with open(os.path.join(models_dir, model_name)) as f:
|
|
model_code = f.read()
|
|
sm = StanModel(model_code=model_code)
|
|
with open(os.path.join(target_dir, target_name), 'wb') as f:
|
|
pickle.dump(sm, f, protocol=pickle.HIGHEST_PROTOCOL)
|
|
|
|
|
|
class BuildPyCommand(build_py):
|
|
"""Custom build command to pre-compile Stan models."""
|
|
|
|
def run(self):
|
|
if not self.dry_run:
|
|
target_dir = os.path.join(self.build_lib, MODELS_TARGET_DIR)
|
|
self.mkpath(target_dir)
|
|
build_stan_models(target_dir)
|
|
|
|
build_py.run(self)
|
|
|
|
|
|
class DevelopCommand(develop):
|
|
"""Custom develop command to pre-compile Stan models in-place."""
|
|
|
|
def run(self):
|
|
if not self.dry_run:
|
|
target_dir = os.path.join(self.setup_path, MODELS_TARGET_DIR)
|
|
self.mkpath(target_dir)
|
|
build_stan_models(target_dir)
|
|
|
|
develop.run(self)
|
|
|
|
|
|
class TestCommand(test_command):
|
|
"""We must run tests on the build directory, not source."""
|
|
|
|
def with_project_on_sys_path(self, func):
|
|
# Ensure metadata is up-to-date
|
|
self.reinitialize_command('build_py', inplace=0)
|
|
self.run_command('build_py')
|
|
bpy_cmd = self.get_finalized_command("build_py")
|
|
build_path = normalize_path(bpy_cmd.build_lib)
|
|
|
|
# Build extensions
|
|
self.reinitialize_command('egg_info', egg_base=build_path)
|
|
self.run_command('egg_info')
|
|
|
|
self.reinitialize_command('build_ext', inplace=0)
|
|
self.run_command('build_ext')
|
|
|
|
ei_cmd = self.get_finalized_command("egg_info")
|
|
|
|
old_path = sys.path[:]
|
|
old_modules = sys.modules.copy()
|
|
|
|
try:
|
|
sys.path.insert(0, normalize_path(ei_cmd.egg_base))
|
|
working_set.__init__()
|
|
add_activation_listener(lambda dist: dist.activate())
|
|
require('%s==%s' % (ei_cmd.egg_name, ei_cmd.egg_version))
|
|
func()
|
|
finally:
|
|
sys.path[:] = old_path
|
|
sys.modules.clear()
|
|
sys.modules.update(old_modules)
|
|
working_set.__init__()
|
|
|
|
|
|
setup(
|
|
name='fbprophet',
|
|
version='0.1.1',
|
|
description='Automatic Forecasting Procedure',
|
|
url='https://facebookincubator.github.io/prophet/',
|
|
author='Sean J. Taylor <sjt@fb.com>, Ben Letham <bletham@fb.com>',
|
|
author_email='sjt@fb.com',
|
|
license='BSD',
|
|
packages=['fbprophet', 'fbprophet.tests'],
|
|
setup_requires=[
|
|
],
|
|
install_requires=[
|
|
'matplotlib',
|
|
'pandas>=0.18.1',
|
|
'pystan>=2.14',
|
|
],
|
|
zip_safe=False,
|
|
include_package_data=True,
|
|
cmdclass={
|
|
'build_py': BuildPyCommand,
|
|
'develop': DevelopCommand,
|
|
'test': TestCommand,
|
|
},
|
|
test_suite='fbprophet.tests',
|
|
long_description="""
|
|
Implements a procedure for forecasting time series data based on an additive model where non-linear trends are fit with yearly and weekly seasonality, plus holidays. It works best with daily periodicity data with at least one year of historical data. Prophet is robust to missing data, shifts in the trend, and large outliers.
|
|
"""
|
|
)
|