saymrwulf/prophet
1
0
Fork 0
mirror of https://github.com/saymrwulf/prophet.git synced 2026-09-16 22:20:23 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Consolidate dataframe validation logic

Browse source
This commit is contained in:
Ben Letham 2017-07-11 16:56:28 -07:00
parent fb517b53d6
commit 796b0f8037
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
python/fbprophet/forecaster.py
View file

@ -180,6 +180,8 @@ class Prophet(object):
""" """
if 'y' in df: if 'y' in df:
df['y'] = pd.to_numeric(df['y']) df['y'] = pd.to_numeric(df['y'])
if np.isinf(df['y'].values).any():
raise ValueError('Found infinity in column y.')
df['ds'] = pd.to_datetime(df['ds']) df['ds'] = pd.to_datetime(df['ds'])
if df['ds'].isnull().any(): if df['ds'].isnull().any():
raise ValueError('Found NaN in column ds.') raise ValueError('Found NaN in column ds.')
@ -554,8 +556,6 @@ class Prophet(object):
raise Exception('Prophet object can only be fit once. ' raise Exception('Prophet object can only be fit once. '
'Instantiate a new object.') 'Instantiate a new object.')
history = df[df['y'].notnull()].copy() history = df[df['y'].notnull()].copy()
if np.isinf(history['y'].values).any():
raise ValueError('Found infinity in column y.')
self.history_dates = pd.to_datetime(df['ds']).sort_values() self.history_dates = pd.to_datetime(df['ds']).sort_values()
history = self.setup_dataframe(history, initialize_scales=True) history = self.setup_dataframe(history, initialize_scales=True)