o U|%i@sdZddlmZmZddlZddlZddlm Z ddl m Z ddl m Z ddlmZddlmZd Zd eeejfd eeejffd d ZGddde e ZdS)zOutlier/anomaly detection utilities for MySQL Connector/Python. Provides a scikit-learn compatible wrapper using HeatWave to score anomalies. )OptionalUnionN) OutlierMixin) MyBaseMLModel)ML_TASK) copy_dict)MySQLConnectionAbstractgh㈵>probreturncCs$t|tdt}t|d|S)z Compute logit (logodds) for a probability, clipping to avoid numerical overflow. Args: prob: Scalar or array of probability values in (0,1). Returns: logit-transformed probabilities. )npclipEPSlog)r resultrP/home/air/sos_test/back/venv/lib/python3.10/site-packages/mysql/ai/ml/outlier.py _get_logits0s rc @seZdZdZ   ddedeedeedeefddZd e e j e j fd e j fd d Zd e e j e j fd e j fd dZd e e j e j fd e j fddZdS)MyAnomalyDetectora MySQL HeatWave scikit-learn compatible anomaly/outlier detector. Flags samples as outliers when the probability of being an anomaly exceeds a user-tunable threshold. Includes helpers to obtain decision scores and anomaly probabilities for ranking. Args: db_connection (MySQLConnectionAbstract): Active MySQL DB connection. model_name (str, optional): Custom model name in the database. fit_extra_options (dict, optional): Extra options for fitting. score_extra_options (dict, optional): Extra options for scoring/prediction. Attributes: boundary: Decision threshold boundary in logit space. Derived from trained model's catalog info Methods: predict(X): Predict outlier/inlier labels. score_samples(X): Compute anomaly (normal class) logit scores. decision_function(X): Compute signed score above/below threshold for ranking. N db_connection model_namefit_extra_optionsscore_extra_optionscCs*tj||tj||dt||_d|_dS)a2 Initialize an anomaly detector instance with threshold and extra options. Args: db_connection: Active MySQL DB connection. model_name: Optional model name in DB. fit_extra_options: Optional extra fit options. score_extra_options: Optional extra scoring options. Raises: ValueError: If outlier_threshold is not in (0,1). DatabaseError: If a database connection issue occurs. If an operational error occurs during execution. )rrN)r__init__rANOMALY_DETECTIONrrboundary)selfrrrrrrrrWs  zMyAnomalyDetector.__init__Xr cCst||dkddS)a Predict outlier/inlier binary labels for input samples. Args: X: Samples to predict on. Returns: ndarray: Values are -1 for outliers, +1 for inliers, as per scikit-learn convention. Raises: DatabaseError: If provided options are invalid or unsupported, or if the model is not initialized, i.e., fit or import has not been called If a database connection issue occurs. If an operational error occurs during execution. DatabaseError: If provided options are invalid or unsupported, or if the model is not initialized, i.e., fit or import has not been called If a database connection issue occurs. If an operational error occurs during execution. gr )r wheredecision_function)rrrrrpredictwszMyAnomalyDetector.predictcCsh||}|jdur/|}|durtd|dddd}|dur(tdtd||_||jS)a Compute signed distance to the outlier threshold. Args: X: Samples to predict on. Returns: ndarray: Score > 0 means inlier, < 0 means outlier; |value| gives margin. Raises: DatabaseError: If provided options are invalid or unsupported, or if the model is not initialized, i.e., fit or import has not been called If a database connection issue occurs. If an operational error occurs during execution. ValueError: If the provided model info does not provide threshold Nz Model does not exist in catalog.model_metadatatraining_paramsanomaly_detection_thresholdzzTrained model is outdated and does not support threshold. Try retraining or using an existing, trained model with MyModel.g?) score_samplesrget_model_info ValueErrorgetr)rr sample_scores model_info thresholdrrrr s    z#MyAnomalyDetector.decision_functioncCs,|jj||jd}t|dddS)aJ Compute normal probability logit score for each sample. Used for ranking, thresholding. Args: X: Samples to score. Returns: ndarray: Logit scores based on "normal" class probability. Raises: DatabaseError: If provided options are invalid or unsupported, or if the model is not initialized, i.e., fit or import has not been called If a database connection issue occurs. If an operational error occurs during execution. )options ml_resultscSs |ddS)N probabilitiesnormalr)xrrrs z1MyAnomalyDetector.score_samples..)_modelr!rrapplyto_numpy)rrrrrrr%s  zMyAnomalyDetector.score_samples)NNN)__name__ __module__ __qualname____doc__rrstrdictrrpd DataFramer ndarrayr!r r%rrrrr>s:    -r)r8typingrrnumpyr pandasr; sklearn.basermysql.ai.ml.basermysql.ai.ml.modelrmysql.ai.utilsrmysql.connector.abstractsrrfloatr=rrrrrrs     &