Aluode/PerceptionLabPortable
0
1o
2 �Yiׇ � @ s� d dl Z d dlZd dlZd dlZd dlZd dlmZ d dl m3Z4 d dl Z d dl m Z m
Z
d dlmZmZmZmZmZmZmZmZ d dlmZ d dlmZ d dlm Z d dl m Z d d l m Z d d5l!m"Z"m#Z# d d l$m%Z% d d l&m'Z' d d
l(m)Z)m*Z* d dl+m,Z,m-Z- d dl.m/Z/ d dl0m1Z1 d dl2m3Z3m4Z4 d dl5m6Z6m7Z7 G dd� de�Z8G dd� de�Z9G dd� de�Z:G dd� de�Z;G dd � d e�Z<G d d � d e;�Z=G d d � d e;e<�Z>G d!d"� d"e>�Z?G d#d$� d$e�Z@G d%d&� d&e�ZAG d'd(� d(�ZBG d)d*� d*e�ZCd+d,� ZDd-d.� ZEd/d0� ZFd1d2� ZGd3d4� ZHd5d6� ZId7d8� ZJd9d:� ZKd;d<� ZLd=d>� ZMd?d@� ZNdAdB� ZOdCdD� ZPejQ�RdEe)� dFfe"e)� dGdHdIgi�dFfe%dJe)� fg�dFfe%dKe"e)� dGdHdIgi�fg�dFfe*� dLfe"e*� dGdHdIgi�dLfe%dMe*� fg�dLfe%dNe"e*� dGdHdIgi�fg�dLfg�dOdP� �ZSejQ�RdEe*� dFfe"e*� dGdHdIgi�dFfe%dMe*� fg�dFfe%dNe"e*� dGdHdIgi�fg�dFfe)� dLfe"e)� dGdHdIgi�dLfe%dJe)� fg�dLfe%dKe"e)� dGdHdIgi�fg�dLfg�dQdR� �ZTejQ�RdEe� dFfe"e� dSdTdUgi�dFfe%dVe� fg�dFfe%dWe"e� dSdTdUgi�fg�dFfe)� dLfe"e)� dGdHdIgi�dLfe%dJe)� fg�dLfe%dKe"e)� dGdHdIgi�fg�dLfg�dXdY� �ZUdZd[� ZVd\d]� ZWd^d_� ZXejQ�Rd`e,dad db�e
jYd dc�fe-dad db�e
jZd dc�fg�ddde� �Z[dfdg� Z\dhdi� Z]djdk� Z^G dldm� dme,�Z_dnZ`dodp� ZaG dqdr� dre,�Zbdsdt� Zcdudv� ZdG dwdx� dx�ZeG dydz� dzeee�Zfd{d|� Zgd}d~� ZhG dd�� d�e�Zid�d�� Zjd�d�� Zkd�d�� Zld�d�� Zmd�d�� Znd�d�� Zod�d�� Zpd�d�� Zqd�d�� Zrd�d�� ZsG d�d�� d��ZtG d�d�� d�ete�ZuejQ�Rd�e� eu� g�d�d�� �Zvd�d�� ZwejQ�Rd�g d���d�d�� �Zxe dFd��d�d�� �Zye dFd��d�d�� �Zzd�d�� Z{d�d�� Z|ejQ�Rd�d�d�dIgfd�e�}dIg�fd�d�dTd�gfd�e�}dTd�g�fd�dg fdd�d�� fej~d�fej~e�}ej~g�fd�d�d�gfd�d�dIdIgfdIe�}dIg�fd�d�d�gfd�e�}d�g�fdIdagdTgfe�}dIg�dadTgfde#� fde d��fg�d�d�� �ZejQ�Rd�d�d�d�g fd�e�}g �fd�d�g d��fd�e�}g d���fej~ej~fd�d�d�d�d�g
�d�dƄ �Z�dS )�� N)�assert_allclose)�config_context�datasets)�
BaseEstimator� OutlierMixin�TransformerMixin�clone�
is_classifier� is_clusterer�is_outlier_detector� is_regressor)�KMeans)�PCA)�IsolationForest)�InconsistentVersionWarning)�6get_scorer)� GridSearchCV�KFold)�Pipeline)�StandardScaler)�SVC�SVR)�DecisionTreeClassifier�DecisionTreeRegressor)�
MockDataFrame)�_get_output_config)�_convert_container�assert_array_equal)�_check_n_features�
validate_datac @ s e Zd Zddd�ZdS )� MyEstimatorr Nc C � || _ || _d S �N��l1�empty)�selfr$ r% � r' ��E:\DocsHouse\542 percep lab latest\PerceptionLab\PerceptionLab_Portable\python_embed\Lib\site-packages\sklearn/tests/test_base.py�__init__/ � 7zMyEstimator.__init__)r N��__name__�8__module__� __qualname__r) r' r' r' r( r . � r c @ � e Zd Zddd�ZdS )�KNc C r! r" )�c�d)r& r2 r3 r' r' r( r) 5 r* z9K.__init__�NNr+ r' r' r' r( r1 4 r/ r1 c @ r0 )�TNc C r! r" )�a�b)r& r6 r7 r' r' r( r) ; r* z10T.__init__r4 r+ r' r' r' r( r5 : r/ r5 c � e Zd Z� fdd�Z� ZS )�NaNTagc � t � �� }d|j_|S )NT��super�__sklearn_tags__�11input_tags� allow_nan�r& �tags�� __class__r' r( r= A � 12zNaNTag.__sklearn_tags__�r, r- r. r= �
__classcell__r' r' rB r( r9 @ � r9 c r8 )�NoNaNTagc r: �NFr; r@ rB r' r( r= H rD zNoNaNTag.__sklearn_tags__rE r' r' rB r( rH G rG rH c r8 )� OverrideTagc r: rI r; r@ rB r' r( r= O rD z OverrideTag.__sklearn_tags__rE r' r' rB r( rJ N rG rJ c @ � e Zd ZdS )�DiamondOverwriteTagN�r, r- r. r' r' r' r( rL U � rL c @ rK )�InheritDiamondOverwriteTagNrM r' r' r' r( rO Y rN rO c @ s$ e Zd ZdZe�dg�fdd�ZdS )�ModifyInitParamsz_Deprecated behavior.13 Equal parameters but with a type cast.14 Doesn't fulfill a is a15 r c C s |� � | _d S r" )�copyr6 �r& r6 r' r' r( r) c � zModifyInitParams.__init__N)r, r- r. �__doc__�np�arrayr) r' r' r' r( rP ] s rP c @ s e Zd ZdZddd�ZdS )�Buggyz9A buggy estimator that does not set its parameters right.Nc C s16 d| _ d S �N� �r6 rR r' r' r( r) j � 17zBuggy.__init__r" �r, r- r. rT r) r' r' r' r( rW g s rW c @ s( e Zd Zdd� Zddd�Zd dd�ZdS )18� NoEstimatorc C � d S r" r' �r& r' r' r( r) o � zNoEstimator.__init__Nc C � | S r" r' �r& �X�yr' r' r( �fitr r` zNoEstimator.fitc C r^ r" r' �r& rc r' r' r( �predictu r` zNoEstimator.predictr4 r" )r, r- r. r) re rg r' r' r' r( r] n s 19r] c @ s e Zd ZdZdd� ZdS )�
VargEstimatorz-scikit-learn estimators shouldn't have vargs.c G r^ r" r' )r& Zvargsr' r' r( r) | r` zVargEstimator.__init__Nr\ r' r' r' r( rh y s rh c C sn ddl m} m} | |dd�}t|�}||usJ �|�� |�� ks"J �| |t�d�d�}t|�}||us5J �d S )Nr �� SelectFpr� f_classif皙�����?��alpha)�20 � )�sklearn.feature_selectionrj rk r �21get_paramsrU Zzeros�rj rk �selectorZ new_selectorr' r' r( �22test_clone� s ru c C s< ddl m} m} | |dd�}d|_t|�}t|d�r J �d S )Nr ri rl rm �test�
own_attribute)rq rj rk rw r �hasattrrs r' r' r( � test_clone_2� s23 ry c C s� t � } d| _t�t�� t| � W d � n1 sw Y t� }t�t�� t|� W d � n1 s6w Y t� }t�t�� t|� W d � n1 sRw Y t � }t�t��
t|� W d � d S 1 sow Y d S )Nrp )24rW r6 �pytest�raises� RuntimeErrorr r] � TypeErrorrh rP )ZbuggyZ no_estimatorZvarg_est�estr' r' r( �test_clone_buggy� s" 25 � 26 � 27 � 28"�r c C s^ t t�g �d�} t| �}t| j|j� t t�t�dgg��d�} t| �}t| jj|jj� d S )N�r% r ) r rU rV r r r% �spZ29csr_matrix�data��clfZclf2r' r' r( �test_clone_empty_array� s r� c C s( t tjd�} t| �}| j|ju sJ �d S �Nr� )r rU �nanr r% r� r' r' r( �test_clone_nan� s r� c C s* dt � i} t| �}| d |d usJ �d S )Nr6 )r r )�origZclonedr' r' r( �test_clone_dict� s 30r� c sn � fdd�t t�D �} | D ]'� � t�d��}t|d�}t|�}|jj|jju s)J �t|j� � |j� � � q
d S )Nc s0 g | ]}|� d �rttt|� � �tu r� �qS )Z_matrix)�endswith�type�getattrr� )�.0�name��clsr' r( �31<listcomp>� s ��z.test_clone_sparse_matrices.<locals>.<listcomp>� r� )32�dirr� rU �eyer r r% rC r Ztoarray)Zsparse_matrix_classesZ
sparse_matrixr� Z33clf_clonedr' r� r( �test_clone_sparse_matrices� s 34�35�r� c C s& t t d�} t| �}| j|ju sJ �d S r� )r r r% r� r' r' r( �test_clone_estimator_types� s 36r� c C s@ d} t jt| d��
tt� W d � d S 1 sw Y d S )Nz8You should provide an instance of scikit-learn estimator��match)rz r{ r} r r )�msgr' r' r( �%test_clone_class_rather_than_instance� s 37"�r� c C sT t � } t| � tt� t� �}t|�dksJ �tdgd d�}tt|��dks(J �d S )NzT(a=K(), b=K())Z long_paramsi� rZ i� )r �reprr5 r1 �len)� my_estimatorrv Zsome_estr' r' r( � test_repr� s r� c C s t � } t| � d S r" )r �str)r� r' r' r( �test_str� s r� c C s� t t� t�} d| jdd�v sJ �d| jdd�vsJ �| jdd� | jjdks(J �t�t�� | jdd� W d � d S 1 s?w Y d S )N�a__dT)�deepFrp )r� )Za__a) r5 r1 rr �38set_paramsr6 r3 rz r{ �39ValueError)rv r' r' r( �test_get_params s "�r� c C s� t jtdd�� tt�sJ �W d � n1 sw Y t jtdd�� tt�s+J �W d � n1 s5w Y t jtdd�� tt�sHJ �W d � n1 sRw Y t jtdd�� t t40�seJ �W d � d S 1 spw Y d S )Nz!passing a class to.*is deprecatedr� ) rz �warns�
FutureWarningr r r r r41 r
r r r' r' r' r( � test_is_estimator_type_class s � � �"�r� zestimator, expected_resultT�Crl rY �svcZsvc_cvFZsvrZsvr_cvc C � t | �|ksJ �d S r" )r �� estimatorZexpected_resultr' r' r( �test_is_classifier � r� c C r� r" )r r� r' r' r( �test_is_regressor0 r� r� Z42n_clusters� � �kmZkm_cvc C r� r" )r43 r� r' r' r( �test_is_clustererA r� r� c C s� t dt� fg�} t�t�� | jdd� W d � n1 s w Y t�t�� | jdd� W d � d S 1 s:w Y d S )Nr� T)Zsvc__stupid_param)Zsvm__stupid_param)r r rz r{ r� r� )r� r' r' r( �test_set_paramsR s � "�r� c sR G � fdd�dt �} ddd�� td| � fg�t| � i �fD ] }|jddd� q d S )Nc s e Zd Z� �fdd�Z� ZS )z?test_set_params_passes_all_parameters.<locals>.TestDecisionTreec s" t � jdi |�� |�ksJ �| S )Nr' )r< r� )r& �kwargs)rC �expected_kwargsr' r( r� i s zJtest_set_params_passes_all_parameters.<locals>.TestDecisionTree.set_params)r, r- r. r� rF r' �r� rB r( �TestDecisionTreeh s r� r� rp )� max_depthZmin_samples_leafr� )Zestimator__max_depthZestimator__min_samples_leaf)r r r r� )r� r~ r' r� r( �%test_set_params_passes_all_parametersd s 4445��r� c C s0 t t� i �} | jt� dd� | jjdksJ �d S )Ng E@)r� Z estimator__C)r r r� r r� r� )Zgscvr' r' r( �$test_set_params_updates_valid_paramsw s r� z tree,datasetrp )r� � random_state)r� c C sh t j�d�}|\}}| �||� |jddt|�d�}| �||�}| j|||d�}d}||ks2J |��d S )Nr rY ro )�size)�
sample_weightz5Unweighted and weighted scores are unexpectedly equal)rU �randomZ RandomStatere �randintr� �score) �treeZdataset�rngrc rd r� Zscore_unweightedZscore_weightedr� r' r' r( �test_score_sample_weight s r� c C sb G dd� dt t�} t�d�}t|�}| |dd�}t|�}|j|jkj�� s'J �|j |j ks/J �d S )Nc @ s, e Zd ZdZd46dd�Zd dd�Zdd � ZdS ) z3test_clone_pandas_dataframe.<locals>.DummyEstimatora, This is a dummy class for generating numerical features47 48 This feature extractor extracts numerical features from pandas data49 frame.50 51 Parameters52 ----------53 54 df: pandas data frame55 The pandas data frame parameter.56 57 Notes58 -----59 NrY c S r! r" )�df� scalar_param)r& r� r� r' r' r( r) � r* z<test_clone_pandas_dataframe.<locals>.DummyEstimator.__init__c S r^ r" r' rb r' r' r( re � r` z7test_clone_pandas_dataframe.<locals>.DummyEstimator.fitc S r^ r" r' rf r' r' r( � transform� r` z=test_clone_pandas_dataframe.<locals>.DummyEstimator.transformrX r" )r, r- r. rT r) re r� r' r' r' r( �DummyEstimator� s60 6162 r� ro rY )r� )63r r rU Zaranger r r� �values�allr� )r� r3 r� �eZcloned_er' r' r( �test_clone_pandas_dataframe� s 64 r� c C s� G dd� dt �} t�ddgddgddgg�}t� �|�}|j}| |�}t|j|� t|�� |�� � t� ddgddgd dgg�}|�|� t|j|� |�65|� t|j|� t |�}||u s`J �t|j|� d66S ) z:Checks that clone works with `__sklearn_clone__` protocol.c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d67� Zd S ) z,test_clone_protocol.<locals>.FrozenEstimatorc S �68 || _ d S r" )�fitted_estimator)r& r� r' r' r( r) � r[ z5test_clone_protocol.<locals>.FrozenEstimator.__init__c S s t | j|�S r" )r� r� )r& r� r' r' r( � __getattr__� s z8test_clone_protocol.<locals>.FrozenEstimator.__getattr__c S ra r" r' r_ r' r' r( �__sklearn_clone__� r` z>test_clone_protocol.<locals>.FrozenEstimator.__sklearn_clone__c _ ra r" r' �r& �argsr� r' r' r( re � r` z0test_clone_protocol.<locals>.FrozenEstimator.fitc _ s | j j|i |��S r" )r� r� r� r' r' r( �
fit_transform� � z:test_clone_protocol.<locals>.FrozenEstimator.fit_transformN)r, r- r. r) r� r� re r� r' r' r' r( �FrozenEstimator� s r� ���������������rp r� � rY N) r rU rV r re Z components_r r Zget_feature_names_outZasarrayr� r )r� rc Zpca�69componentsZ70frozen_pcaZX_newZclone_frozen_pcar' r' r( �test_clone_protocol� s 71 72 r� c C s� t �� } t� �| j| j�}t�|�}d|v sJ �t� � � t�73d� t� |�}W d � n1 s1w Y |� | j| j�}|� | j| j�}||ksLJ �d S )N� _sklearn_version�error)
r � load_irisr re r� �target�pickle�dumps�warnings�catch_warnings� simplefilter�loadsr� )�irisr� Z tree_pickleZ
tree_restoredZscore_of_originalZscore_of_restoredr' r' r( �?test_pickle_version_warning_is_not_raised_with_matching_version� s 74 7576 �r� c @ � e Zd Zdd� ZdS )�TreeBadVersionc C s t | j�� dd�S )N� something)�_sklearn_version)�dict�__dict__�itemsr_ r' r' r( � __getstate__� r� zTreeBadVersion.__getstate__N�r, r- r. r� r' r' r' r( r� � � r� z�Trying to unpickle estimator {estimator} from version {old_version} when using version {current_version}. This might lead to breaking code or invalid results. Use at your own risk.c C s� t �� } t� �| j| j�}t�|�}tj ddt77j d�}t j
t|d��
}t�|� W d � n1 s2w Y |jd j}t|t�sDJ �|jdksKJ �|jdksRJ �|jt78j ksZJ �d S )Nr� r� �r� Z old_version�current_versionr� r )r r� r� re r� r� r� r� �pickle_error_message�format�sklearn� __version__rz r� � UserWarningr� �list�message�79isinstancer Zestimator_nameZoriginal_sklearn_versionZcurrent_sklearn_version)r� r� Ztree_pickle_otherr� Zwarning_recordr' r' r( �<test_pickle_version_warning_is_issued_upon_different_version80 s 81� � r� c @ r� )�
TreeNoVersionc C s | j S r" )r� r_ r' r' r( r� s zTreeNoVersion.__getstate__Nr� r' r' r' r( r� r� r� c C s� t �� } t� �| j| j�}t�|�}d|vsJ �tj ddt82j d�}t j
t|d�� t�|� W d � d S 1 s9w Y d S )Nr� r� zpre-0.18r� r� )r r� r� re r� r� r� r� r� r� r� r� rz r� r� r� )r� r� �tree_pickle_noversionr� r' r' r( �Dtest_pickle_version_warning_is_issued_when_no_version_info_in_pickle" s 83 � "�r c C s� t �� } t� �| j| j�}t�|�}z0tj}dt_t �84� � t � d� t� |� W d � n1 s2w Y W |t_d S W |t_d S |t_w )N�85notsklearnr� )
r r� r� re r� r� r� r� r- r� r� r� r� )r� r� r Z
module_backupr' r' r( �Ctest_pickle_version_no_warning_is_issued_with_non_sklearn_estimator3 s 868788 �89�r c @ s e Zd Zdd� Zdd� ZdS )�DontPickleAttributeMixinc C s | j �� }d |d< |S �N�_attribute_not_pickled)r� rQ )r& r� r' r' r( r� D rD z%DontPickleAttributeMixin.__getstate__c C s d|d< | j �|� d S )NT� _restored)r� �update�r& �stater' r' r( � __setstate__I s z%DontPickleAttributeMixin.__setstate__N)r, r- r. r� r r' r' r' r( r C s r c @ s e Zd Zddd�ZdS )�MultiInheritanceEstimatorr� c C � || _ d | _d S r" ��attribute_pickledr �r& r r' r' r( r) O r* z"MultiInheritanceEstimator.__init__N�r� r+ r' r' r' r( r N r/ r c C sJ t � } d| _t�| �}t�|�}|jdksJ �|jd u s J �|js#J �d S �N�$this attribute should not be pickledr� )r r r� r� r� r r �r� �90serializedZestimator_restoredr' r' r( �3test_pickling_when_getstate_is_overwritten_by_mixinT s 9192r c C s� z<t � } d}|| _t| �j}dt| �_| �� }|d dd�ks J �d|d< | �|� | jdks0J �| js5J �W |t| �_d S |t| �_w )Nr r r� )r r r� r )r r r� r- r� r r r )r� �text�old_modr r' r' r( �Ftest_pickling_when_getstate_is_overwritten_by_mixin_outside_of_sklearn_ s 939495 r c s& e Zd Zddd�Z� fdd�Z� ZS )�SingleInheritanceEstimatorr� c C r
r" r r r' r' r( r) s r* z#SingleInheritanceEstimator.__init__c s t � �� }d |d< |S r )r< r� r rB r' r( r� w rD z'SingleInheritanceEstimator.__getstate__r )r, r- r. r) r� rF r' r' rB r( r r s 96r c C s@ t � } d| _t�| �}t�|�}|jdksJ �|jd u s J �d S r )r r r� r� r� r r r' r' r( �Ctest_pickling_works_when_getstate_is_overwritten_in_the_child_class} s 9798r c C sr t � } t� }| �� jjsJ �|�� jjrJ �t� }|�� jjr!J �t� }|�� jjs,J �t� }|�� jjs7J �d S r" )r9 rH r= r> r? rJ rL rO )Z nan_tag_estZno_nan_tag_estZredefine_tags_estZdiamond_tag_estZinherit_diamond_tag_estr' r' r( �test_tag_inheritance� s r c C sV G dd� dt �} | � }d}tjt|d��
|�� W d � d S 1 s$w Y d S )Nc @ s e Zd Zddd�Zddd�ZdS ) z<test_raises_on_get_params_non_attribute.<locals>.MyEstimatorr� c S r^ r" r' �r& �paramr' r' r( r) � r` zEtest_raises_on_get_params_non_attribute.<locals>.MyEstimator.__init__Nc S ra r" r' rb r' r' r( re � r` z@test_raises_on_get_params_non_attribute.<locals>.MyEstimator.fitr r" )r, r- r. r) re r' r' r' r( r � � 99r z-'MyEstimator' object has no attribute 'param'r� )r rz r{ �AttributeErrorrr )r r~ r� r' r' r( �'test_raises_on_get_params_non_attribute� s 100"�r! c C sv t � } | �� }d|v s
J �d|v sJ �tdd�� | �� }d|v s#J �d|vs)J �W d � d S 1 s4w Y d S )Nz101text/plainz text/htmlr ��display)r �_repr_mimebundle_r )r� �outputr' r' r( �test_repr_mimebundle_� s "�r& c C s� t � } | �� }d|v s
J �tdd��. d}tjt|d�� | �� }W d � n1 s+w Y W d � d S W d � d S 1 sCw Y d S )Nz<style>r r" z _repr_html_ is only defined whenr� )r Z _repr_html_r rz r{ r )r� r% r� r' r' r( �test_repr_html_wraps� s 102 ��"�r' c C sx t � } g d�g d�g}t| |dd� | jdksJ �d}tjt|d�� t| dd d� W d103 � d104S 1 s5w Y d105S ) z>Check that `_check_n_features` validates data when reset=False�rY rp r� )r� r� � T��resetr� zHX does not contain any features, but MyEstimator is expecting 3 featuresr� � invalid XFN)r r �n_features_in_rz r{ r� )r~ ZX_trainr� r' r' r( � test_n_features_in_validation� s "�r. c C s4 t � } t| ddd� t| d�rJ �t| ddd� dS )z]Check that `_check_n_features` does not validate data when106 n_features_in_ is not defined.r, Tr* r- FN)r r rx �r~ r' r' r( � test_n_features_in_no_validation� s r0 c C s� t �d�} t�� }|j}| j||jd�}G dd� dtt�}|� � |�}t107|j |j � |� |� t
|d�r6J �|� |� d}| j||jddd� d�}t jt|d ��
|�|� W d � n1 saw Y d108}t jt|d ��
|�|� W d � n1 sw Y d }|� � |�}t jt|d ��
|�|� W d � n1 s�w Y | �|�}|� }t�� � t�d t� |� |� W d � n1 s�w Y ||g} | D ]!}109t�� � t�d t� |�|110� W d � n1 s�w Y q�| j|g d
�d�} |� }t�d�}t jt|d ��
|� | � W d � n 1 �s w Y t jt|d �� |�| � W d � dS 1 �s>w Y dS )z;Check that feature_name_in are recorded by `_validate_data`�pandas��columnsc @ � e Zd Zddd�Zdd� ZdS )z.test_feature_names_in.<locals>.NoOpTransformerNc S � t | |� | S r" �r rb r' r' r( re � � 111z2test_feature_names_in.<locals>.NoOpTransformer.fitc S s t | |dd� |S �NFr* r6 rf r' r' r( r� � s z8test_feature_names_in.<locals>.NoOpTransformer.transformr" �r, r- r. re r� r' r' r' r( �NoOpTransformer� � 112 r: �feature_names_in_z5The feature names should match those that were passedNr� r� zVX does not have valid feature names, but NoOpTransformer was fitted with feature nameszIX has feature names, but NoOpTransformer was fitted without feature namesr� )r6 r7 rY rp a� Feature names are only supported if all input features have string names, but your input has ['int', 'str'] as feature name / column name types. If you want feature names to be stored and validated, you must convert them all to strings, by using X.columns = X.columns.astype(str) for example. Otherwise you can remove feature / column names from your input data, or convert them all to a non-string data type.)rz � importorskipr r� r� � DataFrame�
feature_namesr r re r r< r3 rx r{ r� r� r� r� r� r� r� �re�escaper} ) �pdr� ZX_npr� r: Ztransr� �df_badZ df_int_namesZXsrc Zdf_mixedr' r' r( �test_feature_names_in� sb 113 114115116 �� � �117118 �119 ��� � $�rD c C s� t �d�} t�� }| j|j|jd�}| �|j�}G dd� dt t120�}|� }t ||dd�}t |t
j�s3J �t||�� � t ||dd�}||u sGJ �t ||dd�}t |t
j�sVJ �t||�� � t ||dd�} | |u sjJ �t |||dd�\}}t |t
j�s|J �t||�� � t |t
j�s�J �t||�� � t |||dd�\}} ||u s�J �| |u s�J �d }121t jt|122d123��
t |� W d � d S 1 s�w Y d S ) z0Check skip_check_array option of _validate_data.r1 r2 c @ rK )z<test_validate_data_skip_check_array.<locals>.NoOpTransformerNrM r' r' r' r( r: 1 rN r: F)�skip_check_arrayT)rd rE z*Validation should be done on X, y or both.r� N)rz r= r r� r> r� r? ZSeriesr� r r r r� rU Zndarrayr Zto_numpyr{ r� ) rB r� r� rd r: �no_opZX_np_outZX_df_outZy_np_outZ y_series_outr� r' r' r( �#test_validate_data_skip_check_array) s8 124 125"�rG c C s: t � jdd�} td| �}t| �}td|�}||ksJ �dS )z-Check that clone keeps the set_output config.r1 )r� r� N)r Z126set_outputr r )�ss�configZss_cloneZ config_cloner' r' r( � test_clone_keeps_output_configR s127 128129rJ c @ rK )�_EmptyNrM r' r' r' r( rK ] rN rK c @ rK )�EmptyEstimatorNrM r' r' r' r( rL a rN rL r� c C s4 | � � }dtji}||ksJ �t�t�t� �� dS )z�Check that ``__getstate__`` returns an empty ``dict`` with an empty130 instance.131 132 Python 3.11+ changed behaviour by returning ``None`` instead of raising an133 ``AttributeError``. Non-regression test for gh-25188.134 r� N)r� r� r� r� r� r� r )r� r135 �expectedr' r' r( �"test_estimator_empty_instance_dicte s 136 rN c C s� G dd� d�} G dd� dt | �}d}tjt|d��
|� �� W d � n1 s)w Y tjt|d�� t�|� � W d � dS 1 sGw Y dS )z:Using a `BaseEstimator` with `__slots__` is not supported.c @ s e Zd ZdZdS )zDtest_estimator_getstate_using_slots_error_message.<locals>.WithSlots��xN)r, r- r. � __slots__r' r' r' r( � WithSlotsx s rR c @ rK )zDtest_estimator_getstate_using_slots_error_message.<locals>.EstimatorNrM r' r' r' r( � Estimator{ rN rS zRYou cannot use `__slots__` in objects inheriting from `sklearn.base.BaseEstimator`r� N)r rz r{ r} r� r� r� )rR rS r� r' r' r( �1test_estimator_getstate_using_slots_error_messageu s � �"�rT z constructor_name, minversion))Z dataframez1.5.0)�pyarrowz12.0.0)Zpolarsz0.20.23c 137 C s� g d�g d�g}g d�}t || ||d�}G dd� dtt�}|� }|�|� t|j|� |�|�}| dkr9t||� g d�}t || |d �} tj t138d139d �� |�| � W d � d S 1 s\w Y d S )
z:Uses the dataframe exchange protocol to get feature names.)rY r� rp )r� r� r) )Zcol_0Zcol_1Zcol_2)� columns_name�140minversionc @ r4 )z0test_dataframe_protocol.<locals>.NoOpTransformerNc S r5 r" r6 rb r' r' r( re � r7 z4test_dataframe_protocol.<locals>.NoOpTransformer.fitc S s t | |dd�S r8 r6 rf r' r' r( r� � rS z:test_dataframe_protocol.<locals>.NoOpTransformer.transformr" r9 r' r' r' r( r: � r; r: rU )r6 r7 r2 )rV z The feature names should matchr� N) r r r re r r< r� r rz r{ r� )141Zconstructor_namerW r� r3 r� r: rF ZX_outZ bad_namesrC r' r' r( �test_dataframe_protocol� s" 142�143 144145 "�rX )Zenable_metadata_routingc C � G dd� dt t�} tjtdd�� | � jdd�jdggdgdd� W d � n1 s+w Y tjdd �� }| � jdd��dggdg� t |�d146ksMJ �W d � dS 1 sXw Y dS ) zkTest that having a transformer with metadata for transform raises a147 warning when calling fit_transform.c @ � e Zd Zddd�Zddd�ZdS )zTtest_transformer_fit_transform_with_metadata_in_transform.<locals>.CustomTransformerNc S ra r" r' �r& rc rd �propr' r' r( re � r` zXtest_transformer_fit_transform_with_metadata_in_transform.<locals>.CustomTransformer.fitc S � |S r" r' �r& rc r\ r' r' r( r� � r` z^test_transformer_fit_transform_with_metadata_in_transform.<locals>.CustomTransformer.transformr4 r" r9 r' r' r' r( �CustomTransformer� r r_ z*`transform` method which consumes metadatar� T�r\ rY N��recordr )148r r rz r� r� Zset_transform_requestr� r� r� r� )r_ rb r' r' r( �9test_transformer_fit_transform_with_metadata_in_transform� � � � "�rc c C rY ) ziTest that having an OutlierMixin with metadata for predict raises a149 warning when calling fit_predict.c @ rZ )zVtest_outlier_mixin_fit_predict_with_metadata_in_predict.<locals>.CustomOutlierDetectorNc S ra r" r' r[ r' r' r( re � r` zZtest_outlier_mixin_fit_predict_with_metadata_in_predict.<locals>.CustomOutlierDetector.fitc S r] r" r' r^ r' r' r( rg � r` z^test_outlier_mixin_fit_predict_with_metadata_in_predict.<locals>.CustomOutlierDetector.predictr4 r" )r, r- r. re rg r' r' r' r( �CustomOutlierDetector� r re z(`predict` method which consumes metadatar� Tr` rY Nra r )150r r rz r� r� Zset_predict_requestZ fit_predictr� r� r� )re rb r' r' r( �7test_outlier_mixin_fit_predict_with_metadata_in_predict� rd rf c C s6 t dd�} | �� ddd�ksJ �| �� jdksJ �dS )z5Check the behaviour of the `_get_params_html` method.rv r� r r# N)r �_get_params_html� non_defaultr/ r' r' r( �test_get_params_html� s 151ri c s G � fdd�dt �}|S )Nc s e Zd Z� fdd�ZdS )z3make_estimator_with_param.<locals>.DynamicEstimatorc S r� r" �r r r' r' r( r) � r[ z<make_estimator_with_param.<locals>.DynamicEstimator.__init__Nr+ r' ��
default_valuer' r( �DynamicEstimator� s rm )r )rl rm r' rk r( �make_estimator_with_param� s rn zdefault_value, test_value)r' )rY r' )�rY rp )r� r� ro r� rX c C ra r" r' rO r' r' r( �<lambda> s rp � �?)�abc�defrr )TF)rq � @Zaccuracyc C s( t | �|d�}|�� j}d|v sJ �dS )z�Check that we detect non-default parameters with various types.152 153 Non-regression test for:154 https://github.com/scikit-learn/scikit-learn/issues/31525155 rj r N�rn rg rh �rl Z156test_valuer� rh r' r' r( �test_param_is_non_default� s #157rw r4 )r' r' )r( r( r( )rr rr )TT)rY rY )rq rq )rp rt c C s( t | �|d�}|�� j}d|vsJ �dS )z�Check that we detect the default parameters and values in an array-like will158 be reported as default as well.159 160 Non-regression test for:161 https://github.com/scikit-learn/scikit-learn/issues/31525162 rj r Nru rv r' r' r( �test_param_is_default s 163rx )�r� r@ r� �numpyrU rz Z scipy.sparse�sparser� Z
numpy.testingr r� r r Z sklearn.baser r r r r r164 r r Zsklearn.clusterr
Zsklearn.decompositionr Zsklearn.ensembler Zsklearn.exceptionsr Zsklearn.metricsr Zsklearn.model_selectionr r Zsklearn.pipeliner Zsklearn.preprocessingr Z sklearn.svmr r Z sklearn.treer r Zsklearn.utils._mockingr Zsklearn.utils._set_outputr Zsklearn.utils._testingr r Zsklearn.utils.validationr r r r1 r5 r9 rH rJ rL rO rP rW r] rh ru ry r r� r� r� r� r� r� r� r� r� r� �markZ parametrizer� r� r� r� r� r� Zmake_classificationZmake_regressionr� r� r� r� r� r� r� r� r r r r r r r r r r! r& r' r. r0 rD rG rJ rK rL rN rT rX rc rf ri rn rV r� rw rx r' r' r' r( �<module> s| ( 165 166 ��167
��168
��169
170171�172173���174
&,� 175
N) 176�177 178179180181 182��183 184185��