o U|%i@sddZddlmZmZddlZddlmZddlm Z ddl m Z m Z ddl mZGdd d eZdS) zBase classes for MySQL HeatWave ML estimators for Connector/Python. Implements a scikit-learn-compatible base estimator wrapping server-side ML. )OptionalUnionN) BaseEstimator)MySQLConnectionAbstract)ML_TASKMyModel) copy_dictc @seZdZdZ  ddedeeefdeedee fddZ dd e j d ee j d dfd d Z d efddZd ee fddZdS) MyBaseMLModela Base class for MySQL HeatWave machine learning estimators. Implements the scikit-learn API and core model management logic, including fit, explain, serialization, and dynamic option handling. For use as a base class by classifiers, regressors, transformers, and outlier models. Args: db_connection (MySQLConnectionAbstract): An active MySQL connector database connection. task (str): ML task type, e.g. "classification" or "regression". model_name (str, optional): Custom name for the deployed model. fit_extra_options (dict, optional): Extra options for fitting. Attributes: _model: Underlying database helper for fit/predict/explain. fit_extra_options: User-provided options for fitting. N db_connectiontask model_namefit_extra_optionscCst|||d|_t||_dS)a Initialize a MyBaseMLModel with connection, task, and option parameters. Args: db_connection: Active MySQL connector database connection. task: String label of ML task (e.g. "classification"). model_name: Optional custom model name. fit_extra_options: Optional extra fit options. Raises: DatabaseError: If a database connection issue occurs. If an operational error occurs during execution. )r r N)r_modelrr )selfr r r r rM/home/air/sos_test/back/venv/lib/python3.10/site-packages/mysql/ai/ml/base.py__init__?szMyBaseMLModel.__init__XyreturncCs|j|||j|S)aZ Fit the underlying ML model using pandas DataFrames. Delegates to MyMLModelPandasHelper.fit. Args: X: Features DataFrame. y: (Optional) Target labels DataFrame or Series. Returns: self Raises: DatabaseError: If provided options are invalid or unsupported. If a database connection issue occurs. If an operational error occurs during execution. Notes: Additional temp SQL resources may be created and cleaned up during the operation. )rfitr )rrrrrrrWszMyBaseMLModel.fitcC |jS)a% Deletes the model from the model catalog if present Raises: DatabaseError: If a database connection issue occurs. If an operational error occurs during execution. Returns: Whether the model was deleted )r _delete_modelrrrrrss zMyBaseMLModel._delete_modelcCr)a Checks if the model name is available. Model info will only be present in the catalog if the model has previously been fitted. Returns: True if the model name is not part of the model catalog Raises: DatabaseError: If a database connection issue occurs. If an operational error occurs during execution. )rget_model_inforrrrrs zMyBaseMLModel.get_model_info)NN)N)__name__ __module__ __qualname____doc__rrstrrrdictrpd DataFramerboolrrrrrrr ,s.   r )rtypingrrpandasr! sklearn.basermysql.connector.abstractsrmysql.ai.ml.modelrrmysql.ai.utilsrr rrrrs