o 0jy@s\ddlmZmZmZddlZddlZddlm Z e j GdddZ e j GdddZ dS) )AnyDictListN) benchmarkcsPeZdZdZfddZdeedeejfddZ dedejfd d Z Z S) GetClszFA class to process prediction outputs and return class IDs and scores.cstdS)z Initializes the GetCls instance.N)super__init__self __class__v/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddlex/inference/models/ts_classification/processors.pyr szGetCls.__init__ pred_listreturncfdd|DS)a Processes a list of predictions and returns a list of DataFrames with class IDs and scores. Args: pred_list (List[Any]): A list of predictions, where each prediction is expected to be an iterable of arrays. Returns: List[pd.DataFrame]: A list of DataFrames, each containing the class ID and score for the corresponding prediction. cg|]}|qSr)getcls).0predr rr )z#GetCls.__call__..r)r rrr r__call__s zGetCls.__call__rcCst|d}|tj|ddd8}t|tjt|ddd}tj|dd}||}tj|g|gd}d|j_ |S)aS Computes the class ID and score from a single prediction. Args: pred (Any): A prediction, expected to be an iterable where the first element is an array representing logits or probabilities. Returns: pd.DataFrame: A DataFrame containing the class ID and score for the prediction. rT)axisZkeepdims)r)classidZscoresample) npmaxexpsumZargmaxpd DataFrame from_dictindexname)r rZpred_tsrZ pred_scoreresultrrrr+s  z GetCls.getcls) __name__ __module__ __qualname____doc__r rrr"r#rr __classcell__rrr rrs   rcszeZdZdZdeeefffdd ZdeeeefdeeeeffddZ d eeefdeeeffd d Z Z S) BuildPadMaskz4A class to build padding masks for time series data. input_datacst||_dS)a Initializes the BuildPadMask instance. Args: input_data (Dict[str, Any]): A dictionary containing configuration data, including 'features' and 'pad_mask' keys that influence how padding is applied. N)rr r.)r r.r rrr Cs  zBuildPadMask.__init__ts_listrcr)a Applies padding mask to a list of time series data. Args: ts_list (List[Dict[str, Any]]): A list of dictionaries, each representing a time series instance with keys like 'features' and 'past_target'. Returns: List[Dict[str, Any]]: A list of dictionaries with updated 'features' and 'pad_mask' keys. crr)padmask)rtsr rrrYrz)BuildPadMask.__call__..r)r r/rr rrNs zBuildPadMask.__call__r1cCsd|jvr |d|d<d|jvret|d}|jdd}|dkretj|tjd}||krat|dtj}tj||gtjd}t ||}||d|ddf<||d<d||d<||d<|S||d<|S) a Builds a padding mask for a single time series instance. Args: ts (Dict[str, Any]): A dictionary representing a time series instance, expected to have keys like 'features' and 'past_target'. Returns: Dict[str, Any]: The input dictionary with potentially updated 'features' and 'pad_mask' keys. featuresZ past_targetZpad_maskrr)ZdtypeNg) r.lenronesZint32arrayZastypeZfloat32Zzerosmin)r r1Z target_dim max_lengthr4Ztarget_ndarrayZtarget_ndarray_finalendrrrr0[s(      zBuildPadMask.padmask) r(r)r*r+rstrrr rrr0r,rrr rr-?s * * r-) typingrrrnumpyrZpandasr"Zutils.benchmarkrZtimeitrr-rrrrs '