forecastflowml.ForecastFlowML.predict#

ForecastFlowML.predict(df, trained_models=None, spark=None)[source]#

Make predictions

Parameters:

  • df (pyspark.sql.dataframe.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 local_result=True during training.

Return type:

DataFrame that contains predictions per time series.