a
    )a                     @   s  d dl Zd dlmZ d dlmZmZ d dlmZm	Z	m
Z
 d dlmZmZmZmZmZmZ d dlmZmZ d dlmZ d dlmZ d d	lmZ ed
dg ddde	d
dg ddde
d
ddgdde dgddeddg ddde g ddde ddgdde ddgdde g dddeddg dddeddd g d!ddedd"g d!ddeddddgdde ddgdde ddgddgZejejejejej ej!ej"ej#ej$ej%ej&gZ'd#d$d%d&d'd(d)d*d+d,d-d.d/d0ed1sd2ndgZ(dS )3    N)SVC)RandomForestClassifierRandomForestRegressor)KNeighborsClassifierKNeighborsRegressorNearestNeighbors)LogisticRegressionLogisticRegressionCVLinearRegressionRidge
ElasticNetLasso)KMeansDBSCAN)TSNE)PCA)daal_check_versionZbrute)	algorithm)
kneighborspredictpredict_probascoreZ
classifier)ZmodelmethodsZdataset)r   r   r   Z
regressionr   Zblobsfit_predictT)Zprobability)decision_functionr   r   r   )r   fit_transform	transformr   r   r   r   )r   r   r   
   )Zn_estimators)r   r   predict_log_probar   d   Zmultinomial)max_iterZmulti_class)r   r   r   r   r   )r    zKMeans .*transformzKMeans .*scorezPCA .*scorez&LogisticRegression .*decision_functionz(LogisticRegressionCV .*decision_functionzLogisticRegressionCV .*predictz$LogisticRegressionCV .*predict_probaz(LogisticRegressionCV .*predict_log_probazLogisticRegressionCV .*scorezRidge float16 predictzRidge float16 scorez&RandomForestClassifier .*predict_probaz*RandomForestClassifier .*predict_log_probaz'pairwise_distances .*pairwise_distances)i  P   zroc_auc_score .*roc_auc_score))ZnumpyZnpZsklearn.svmr   Zsklearn.ensembler   r   Zsklearn.neighborsr   r   r   Zsklearn.linear_modelr   r	   r
   r   r   r   Zsklearn.clusterr   r   Zsklearn.manifoldr   Zsklearn.decompositionr   Zdaal4py.sklearn._utilsr   ZMODELS_INFOZint8Zint16Zint32Zint64Zfloat16Zfloat32Zfloat64Zuint8Zuint16Zuint32Zuint64ZTYPESZTO_SKIP r#   r#   Mlib/python3.9/site-packages/daal4py/sklearn/monkeypatch/tests/_models_info.py<module>   s    
Q