KidZddlmZmZddlZddlZddlm Z ddl m Z ddl m Z ddlmZddlmZGd d e e ZdS) zwClassifier utilities for MySQL Connector/Python. Provides a scikit-learn compatible classifier backed by HeatWave ML. )OptionalUnionN)ClassifierMixin) MyBaseMLModel)ML_TASK) copy_dict)MySQLConnectionAbstractc "eZdZdZ ddedeedeedeedeef dZd e e j e j fd e j fd Zd e e j e j fd e j fd Zd e e j e j fd e j fd ZdS) MyClassifiera MySQL HeatWave scikit-learn compatible classifier estimator. Provides prediction and probability output from a model deployed in MySQL, and manages fit, explain, and prediction options as per HeatWave ML interface. Attributes: predict_extra_options (dict): Dictionary of optional parameters passed through to the MySQL backend for prediction and probability inference. _model (MyModel): Underlying interface for database model operations. fit_extra_options (dict): See MyBaseMLModel. Args: db_connection (MySQLConnectionAbstract): Active MySQL connector DB connection. model_name (str, optional): Custom name for the model. fit_extra_options (dict, optional): Extra options for fitting. explain_extra_options (dict, optional): Extra options for explanations. predict_extra_options (dict, optional): Extra options for predict/predict_proba. Methods: predict(X): Predict class labels. predict_proba(X): Predict class probabilities. N db_connection model_namefit_extra_optionsexplain_extra_optionspredict_extra_optionsctj||tj||t ||_t ||_dS)a Initialize a MyClassifier. Args: db_connection: Active MySQL connector database connection. model_name: Optional, custom model name. fit_extra_options: Optional fit options. explain_extra_options: Optional explain options. predict_extra_options: Optional predict/predict_proba options. Raises: DatabaseError: If a database connection issue occurs. If an operational error occurs during execution. )r rN)r__init__rCLASSIFICATIONrrr)selfr r rrrs R/var/www/html/analyses/venv/lib/python3.11/site-packages/mysql/ai/ml/classifier.pyrzMyClassifier.__init__GsZ.     "!/     &//D%E%E"%./D%E%E"""Xreturncx|j||j}|dS)a Predict class labels for the input features using the MySQL model. References: https://dev.mysql.com/doc/heatwave/en/mys-hwaml-ml-predict-table.html A full list of supported options can be found under "ML_PREDICT_TABLE Options" Args: X: Input samples as a numpy array or pandas DataFrame. Returns: ndarray: Array of predicted class labels, shape (n_samples,). 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 Prediction)_modelpredictrto_numpy)rrresults rrzMyClassifier.predicths80$$Q0J$KKl#,,...rc|j||j}t|djddt j|dfdS)a* Predict class probabilities for the input features using the MySQL model. References: https://dev.mysql.com/doc/heatwave/en/mys-hwaml-ml-predict-table.html A full list of supported options can be found under "ML_PREDICT_TABLE Options" Args: X: Input samples as a numpy array or pandas DataFrame. Returns: ndarray: Array of shape (n_samples, n_classes) with class probabilities. 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. r ml_resultsr probabilitiesc"fdDS)Nc,g|]}d|S)r#).0 class_name ml_results r z@MyClassifier.predict_proba....s0###?IIo.z:###rr&)r)classess`rz,MyClassifier.predict_proba..s,####MT###r) rrrsortedilockeysnpstackmap)rrr r+s @r predict_probazMyClassifier.predict_probas0$$Q0J$KK -215oFKKMMNNx < $ $     rcH|j||jdS)ai Explain model predictions using provided data. References: https://dev.mysql.com/doc/heatwave/en/mys-hwaml-ml-explain-table.html A full list of supported options can be found under "ML_EXPLAIN_TABLE Options" Args: X: DataFrame for which predictions should be explained. Returns: DataFrame containing explanation details (feature attributions, etc.) 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. Notes: Temporary input/output tables are cleaned up after explanation. rN)rexplain_predictionsr)rrs rr5z MyClassifier.explain_predictionss(6 ''43M'NNNNNr)NNNN)__name__ __module__ __qualname____doc__r rstrdictrrpd DataFramer0ndarrayrr3r5r&rrr r .s56%),00404 FF.FSMF$D> F (~ F (~ FFFFB/r|RZ/0/ ////6" r|RZ/0" " " " " HOr|RZ/0O OOOOOOrr )r9typingrrnumpyr0pandasr< sklearn.basermysql.ai.ml.basermysql.ai.ml.modelrmysql.ai.utilsrmysql.connector.abstractsr r r&rrrGs:#"""""""((((((******%%%%%%$$$$$$======TOTOTOTOTO=/TOTOTOTOTOr