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���|�dZd
dl
mZmZ
d
dlZd
dlZd
dlm Z
d
dl mZ
d
dlm Z
d
dlmZ
d
dlmZ
Gd �d ee �Zy)zwClassifier utilities for MySQL Connector/Python. Provides a scikit-learn compatible classifier backed by HeatWave ML. �)�Optional�UnionN)
�ClassifierMixin)
� MyBaseMLModel)
�ML_TASK)
� copy_dict)
�MySQLConnectionAbstractc�P
�eZ
dZd
Z ddedeedeedeedeef d�Zd e e jejfd ejfd�Zd e e jejfd ejfd�Zd e e jejfd e jfd �Zy)�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_name�fit_extra_options�explain_extra_options�predict_extra_optionsc��t
j||
tj||�
�
t |�
|_t |�
|_y)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__r�CLASSIFICATIONrrr)�selfrr rrrs �G/opt/nydus/tmp/pip-target-a90h98xg/lib/python/mysql/ai/ml/classifier.pyrzMyClassifier.__init__GsE��. ������"�"�!�/� �&/�/D�%E��"�%.�/D�%E��"��
X�returnc�v�|jj|
|j�
�}|dj�S)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)�_model�predictr�to_numpy)rr�results rrzMyClassifier.predicths7��0���$�$�Q��0J�0J�$�K���l�#�,�,�.�.rc����|jj|
|j�
�}t|djddj��
�t j|dj�f
d��
�
S)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
�:�
��D�
cgc] }
|d
|
��c}
Sc
c}
w)Nr#�)� ml_result� class_name�classess �r�z,MyClassifier.predict_proba..�s)���MT�#�MT�z�I�o�.�z�:�W�#��#s�) rrr�sorted�iloc�keys�np�stack�map)rrr r(s @r� predict_probazMyClassifier.predict_proba�su���0���$�$�Q��0J�0J�$�K�����-�2�2�1�5�o�F�K�K�M�N���x�x��<� �$�$�� � � rc�R�|jj|
|j�
�
y)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)r�explain_predictionsr)rrs rr2z MyClassifier.explain_predictions�s!��6 ���'�'�
�4�3M�3M�'�Nr)NNNN)�__name__� __module__�__qualname__�__doc__r r�str�dictrr�pd� DataFramer-�ndarrayrr0r2r%rrrr.s����6%)�,0�04�04� F
�.�F
��S�M�F
�$�D�>� F
� (��~�F
� (��~� F
�B
/��r�|�|�R�Z�Z�/�0�/� ���/�6" ��r�|�|�R�Z�Z�/�0�" � ���" �H
O
��r�|�|�R�Z�Z�/�0�O
� ���O
rr)r6�typingrr�numpyr-�pandasr9�sklearn.baser�mysql.ai.ml.baser�mysql.ai.ml.modelr�mysql.ai.utilsr�mysql.connector.abstractsr rr%rr�rD
s6�


�:
�
#���(�*�%�$�=�T
O
�=�/�T
O
r