o 0j @sddlZddlmZddlmZddlmZmZddlm Z edr(ddl m Z ddl Zedd ejd ejfd d ZGd dde ZdS)N)Any)Image)function_requires_depsis_dep_available) BaseTSResultZ matplotlibforecastreturncCstjdd|j}|jj|jt|_tj|j||dddddtdt d t |dt t d tj tdt|d d tj d dt}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. Returns: Image.Image: The visualized result as a Pillow image. ) )ZfigsizerpostZAnomalyred)wherelabelcolorzTime Series Anomaly DetectionZTimeT )Zticks-)ZrotationZtight)Z bbox_inches)pltZfigurecolumnsindexnameZastypestrsteptitleZxlabelZylabelZlegendgridZxticksrangelenioBytesIOZsavefigseekcloseropen)r Zforecast_columnsbufimager%u/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddlex/inference/models/ts_anomaly_detection/result.py visualizes0        r'c@s.eZdZdZdejfddZdefddZdS) TSAdResultzHA class representing the result of a time series anomaly detection task.r cCs|d}dt|iS)applyanomalyres)r')selfr*r%r%r&_to_imgGs zTSAdResult._to_imgcCs d|diS)z Converts the anomaly detection results to a CSV format. Returns: Any: The anomaly data formatted for CSV output, typically a DataFrame or similar structure. r+r*r%)r,r%r%r&_to_csvLs zTSAdResult._to_csvN)__name__ __module__ __qualname____doc__rr-rr.r%r%r%r&r(Dsr()rtypingrZPILrZ utils.depsrrZ common.resultrZmatplotlib.pyplotZpyplotrZpandaspdZ DataFramer'r(r%r%r%r&s    '