saymrwulf/prophet
1
0
Fork 0
mirror of https://github.com/saymrwulf/prophet.git synced 2026-05-17 21:10:45 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix bug in performance metrics

Browse source
This commit is contained in:
Alexander Gawrilow 2019-07-31 15:31:26 +02:00 committed by Ben Letham
parent 7f214f2bc7
commit c179a28be2
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
python/fbprophet/diagnostics.py
View file

@ -238,7 +238,7 @@ def performance_metrics(df, metrics=None, rolling_window=0.1):
valid_metrics = ['mse', 'rmse', 'mae', 'mape', 'mdape', 'coverage']
if metrics is None:
metrics = valid_metrics
if ('yhat_lower' not in df) or ('yhat_upper' not in df) and ('coverage' in metrics):
if ('yhat_lower' not in df or 'yhat_upper' not in df) and ('coverage' in metrics):
metrics.remove('coverage')
if len(set(metrics)) != len(metrics):
raise ValueError('Input metrics must be a list of unique values')