✏️ 正在编辑: base.cpython-312.pyc
路径:
/var/opt/nydus/ops/mysql/ai/ml/__pycache__/base.cpython-312.pyc
提示:
您可以编辑任何文件（包括二进制文件），但请注意不当修改可能导致文件损坏。
� ��'j� � �j � d Z ddlmZmZ ddlZddlmZ ddlm Z ddl m Z m Z ddl mZ G d� 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 � � e Zd ZdZ ddedeeef dee dee fd�Z dde j d ee j d d fd �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 r N)r �_modelr r )�selfr r r r s �A/opt/nydus/tmp/pip-target-k0fmikie/lib/python/mysql/ai/ml/base.py�__init__zMyBaseMLModel.__init__? s" � �* �m�$�:�N�� �!*�+<�!=�� � �X�y�returnc �T � | j 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. )r �fitr )r r r s r r zMyBaseMLModel.fitW s# � �2 � � ����1�d� 4� 4�5�� r c �6 � | j j � 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�r s r r zMyBaseMLModel._delete_models s � � �{�{�(�(�*�*r c �6 � | j j � 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_infor s r r z MyBaseMLModel.get_model_info� s � � �{�{�)�)�+�+r )NN)N)�__name__� __module__� __qualname__�__doc__r r �strr r �dictr �pd� DataFramer �boolr r � r r r r , s� � ��, %)�,0� >�.�>� �C��L�!�>� �S�M� >� $�D�>� >�6 %)�� �<�<�� �B�L�L� !�� � �8 +�t� +� ,���� ,r r )r# �typingr r �pandasr& � sklearn.baser �mysql.connector.abstractsr �mysql.ai.ml.modelr r �mysql.ai.utilsr r r) r r �<module>r0 s, ��:� #� � &� =� .� $�b,�M� b,r
💾 保存文件
← 返回文件管理器