forecastflowml.ForecastFlowML.predict

ForecastFlowML.predict(df, trained_models=None, spark=None)

Make predictions

Parameters:

• df (pandas.core.frame.DataFrame) – Dataset to perform predictions on.

• trained_models – pyspark DataFrame that contains the trained models. Does not need to be provided in case local_result is set to True during training.

• spark – Spark session instance. Only provide when df is a pandas DataFrame.

Return type:

DataFrame that contains predictions per time series.