5j|dZddlmZmZddlZddlZddlm Z ddl m Z ddl m Z ddlmZddlmZGd d e e Zy) zuRegressor utilities for MySQL Connector/Python. Provides a scikit-learn compatible regressor backed by HeatWave ML. )OptionalUnionN)RegressorMixin) MyBaseMLModel)ML_TASK) copy_dict)MySQLConnectionAbstractc eZdZdZ d dedeedeedeedeef dZd e e je jfd e jfd Zd e e je jfd e jfd Zy) MyRegressora MySQL HeatWave scikit-learn compatible regressor estimator. Provides prediction output from a regression model deployed in MySQL, and manages fit, explain, and prediction options as per HeatWave ML interface. Attributes: predict_extra_options (dict): Optional parameter dict passed to the backend for prediction. _model (MyModel): Underlying interface for database model operations. fit_extra_options (dict): See MyBaseMLModel. explain_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 predictions. Methods: predict(X): Predict regression target. N db_connection model_namefit_extra_optionsexplain_extra_optionspredict_extra_optionsctj||tj||t ||_t ||_y)a  Initialize a MyRegressor. 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 prediction options. Raises: DatabaseError: If a database connection issue occurs. If an operational error occurs during execution. )r rN)r__init__r REGRESSIONrrr)selfr r rrrs T/var/www/html/serviGia/entorno/lib/python3.12/site-packages/mysql/ai/ml/regressor.pyrzMyRegressor.__init__FsE.      !/  &//D%E"%./D%E"Xreturncv|jj||j}|djS)aZ Predict a continuous target for the input features using the MySQL model. Args: X: Input samples as a numpy array or pandas DataFrame. Returns: ndarray: Array of predicted target values, 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 rrzMyRegressor.predicths7($$Q0J0J$Kl#,,..rcR|jj||jy)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 rr"zMyRegressor.explain_predictionss!6 ''43M3M'Nr)NNNN)__name__ __module__ __qualname____doc__r rstrdictrrpd DataFramenpndarrayrr"rrr r .s4%),00404 F. FSM F$D> F (~ F (~ FD/r||RZZ/0/ /.Or||RZZ/0O Orr )r&typingrrnumpyr+pandasr) sklearn.basermysql.ai.ml.basermysql.ai.ml.modelrmysql.ai.utilsrmysql.connector.abstractsr r r-rrr6s6:#'*%$=lO-lOr