o 0j~ @sddlZddlmZddlZddlmZddlmZm Z ddl m Z e dr,ddl m Zedd ejd ejd ejfd d ZGddde ZdS)N)Any)Image)function_requires_depsis_dep_available) BaseTSResultZ matplotlibforecast actual_datareturncCs$tjdd|j}|jj}||}|jt|_|jt|_tt |t |}| |}tj |j||dddddtj |j||ddd d t d t d t|dttd tjtdd|ddtjddt}tj|dd|dtt|}|S)a Visualizes both the time series forecast and actual results, returning them as a Pillow image. Args: forecast (pd.DataFrame): The DataFrame containing the forecast data. actual_data (pd.Series): The actual observed data for comparison. title (str): The title of the plot. Returns: Image.Image: The visualized result as a Pillow image. ) )Zfigsizerz Actual Databluez--)labelcolorZ linestyleZForecastred)rrzTime Series ForecastZTimeT )Zticks-)ZrotationZtight)Z bbox_inches)pltZfigurecolumnsindexnameZ set_indexZastypestrminlentailZplottitleZxlabelZylabelZlegendgridZxticksrangeioBytesIOZsavefigseekcloseropen)r r Zforecast_columnsZ index_namelengthbufimager(o/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddlex/inference/models/ts_forecasting/result.py visualizes>          r*c@s.eZdZdZdejfddZdefddZdS) TSFcResultzBA class representing the result of a time series forecasting task.r cCs|d}|d}dt||iS)applyr Z cutoff_tsres)r*)selfr Zts_inputr(r(r)_to_imgSszTSFcResult._to_imgcCs d|diS)z Converts the forecasting results to a CSV format. Returns: Any: The forecast data formatted for CSV output, typically a DataFrame or similar structure. r-r r()r.r(r(r)_to_csvYs zTSFcResult._to_csvN)__name__ __module__ __qualname____doc__rr/rr0r(r(r(r)r+Psr+)r typingrZpandaspdZPILrZ utils.depsrrZ common.resultrZmatplotlib.pyplotZpyplotrZ DataFramer*r+r(r(r(r)s    3