fbprophet

00-fbprophet

00-fbprophet

01_fbprophet_getting_started.py

################################################################################################
#	name:	01_fbprophet_getting_started.py
#	desc:	Official tutorial
#	date:	NA
#	Author:	NA
################################################################################################
import pandas as pd
from fbprophet import Prophet

print('*** Program Started ***')
### Impoting data as pandas dataframe
df = pd.read_csv('example_wp_log_peyton_manning.csv')

print(df.head())

### Initiating new object Prophet & initiating fit method and passing input dataframe. Fitting should take 1-5 seconds
m = Prophet()
m.fit(df)

print("type of m" , type(m))

### Extending data to few future dates
future = m.make_future_dataframe(periods=365)
print("type of future" , type(future))


### The predict method will assign each row in future a predicted value which it names yhat
forecast = m.predict(future)
print("type of forecast" , type(future))
# print(forecast[['ds', 'yhat', 'yhat_lower', 'yhat_upper']].tail())

### Plotting forecast
fig1 = m.plot(forecast)
# fig1.show()
fig1.savefig('01_fbprophet_getting_started-01.png')

### Plotting forecast components
fig2 = m.plot_components(forecast)
fig2.savefig('01_fbprophet_getting_started-02.png')

### Saving output excel
forecast.to_csv('example_wp_log_peyton_manning_output.csv', sep=',')

print('*** Program Completed ***')