Name	Size	Mode	Actions
base.cpython-312.pyc	4786	0644	editdlrm
classifier.cpython-312.pyc	7547	0644	editdlrm
model.cpython-312.pyc	30534	0644	editdlrm
outlier.cpython-312.pyc	8048	0644	editdlrm
regressor.cpython-312.pyc	5470	0644	editdlrm
transformer.cpython-312.pyc	6106	0644	editdlrm
__init__.cpython-312.pyc	764	0644	editdlrm

� N�j���j�dZd
dl
mZmZ
d
dlZd
dlmZ
d
dlm Z
d
dl mZmZ
d
dl mZ
Gd�d e�Zy) z�Base classes for MySQL HeatWave ML estimators for Connector/Python. Implements a scikit-learn-compatible base estimator wrapping server-side ML. �)�Optional�UnionN)
� BaseEstimator)
�MySQLConnectionAbstract)�ML_TASK�MyModel)
� copy_dictc ��eZ
dZd
Z ddedeeefdeedee fd�Z ddejd eejd dfd�Z d efd�Zd ee fd �Zy)� 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_connection�task� model_name�fit_extra_optionsc�J�t
|
||�
�|_
t|�
|_y)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 rN)r�_modelr r)�selfrr rrs �A/opt/nydus/tmp/pip-target-a90h98xg/lib/python/mysql/ai/ml/base.py�__init__zMyBaseMLModel.__init__?s"��*�m�$�:�N���!*�+<�!=����
X�
y�returnc�T�|jj|
||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. )r�fitr)rrrs rrzMyBaseMLModel.fitWs#��2 �����
�1�d�4�4�5��rc
�6�|jj�S)
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_model�
rs
rrzMyBaseMLModel._delete_modelss���{�{�(�(�*�*rc
�6�|jj�S)
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. )r�get_model_infors
rrzMyBaseMLModel.get_model_info�s���{�{�)�)�+�+r)NN)
N)�__name__� __module__�__qualname__�__doc__rr�strrr�dictr�pd� DataFramer�boolrr�rrrr,s����,%)�,0�>�.�>��C��L�!�>��S�M� >� $�D�>�>�6%)�� �<�<���B�L�L�!�� � �8+�t�+� ,���� ,rr)r#�typingrr�pandasr&�sklearn.baser�mysql.connector.abstractsr�mysql.ai.ml.modelrr�mysql.ai.utilsr rr)rr�r0
s,�


�:
�
#��&�=�.�$�b

,�M�b

,r