h"ddlmZddlmZddlmZddlmZddlm Z dd d Z dd Z d Z d Z dZGddZdddZdS)wraps)issparse)check_pandas_support) get_config) available_ifN)indexc&t|rtdt|r |}n#t$rd}YnwxYwt d}t ||jr |||_|S||||S)aACreate a Pandas DataFrame. If `data_to_wrap` is a DataFrame, then the `columns` and `index` will be changed inplace. If `data_to_wrap` is a ndarray, then a new DataFrame is created with `columns` and `index`. Parameters ---------- data_to_wrap : {ndarray, dataframe} Data to be wrapped as pandas dataframe. columns : callable, ndarray, or None The column names or a callable that returns the column names. The callable is useful if the column names require some computation. If `columns` is a callable that raises an error, `columns` will have the same semantics as `None`. If `None` and `data_to_wrap` is already a dataframe, then the column names are not changed. If `None` and `data_to_wrap` is **not** a dataframe, then columns are `range(n_features)`. index : array-like, default=None Index for data. `index` is ignored if `data_to_wrap` is already a DataFrame. Returns ------- dataframe : DataFrame Container with column names or unchanged `output`. z+Pandas output does not support sparse data.Nz$Setting output container to 'pandas')r columns)r ValueErrorcallable Exceptionr isinstance DataFramer ) data_to_wrapr r pds U/var/www/html/Sam_Eipo/venv/lib/python3.11/site-packages/sklearn/utils/_set_output.py_wrap_in_pandas_containerr sD HFGGG giiGG   GGG  D E EB, --  #*L  << E7< C CCs : A A ct|di}||vr ||}nt|d}|dvrtd|d|iS)aGet output config based on estimator and global configuration. Parameters ---------- method : {"transform"} Estimator's method for which the output container is looked up. estimator : estimator instance or None Estimator to get the output configuration from. If `None`, check global configuration is used. Returns ------- config : dict Dictionary with keys: - "dense": specifies the dense container for `method`. This can be `"default"` or `"pandas"`. _sklearn_output_config_output>pandasdefaultz0output config must be 'default' or 'pandas' got dense)getattrr r)method estimatorest_sklearn_output_config dense_configs r_get_output_configr"?s|(!( 3KR P P ***08 !||v$6$6$67 000 M| M M    \ ""ct||}|ddkst|s|St|t|dd|jS)aWrap output with container based on an estimator's or global config. Parameters ---------- method : {"transform"} Estimator's method to get container output for. data_to_wrap : {ndarray, dataframe} Data to wrap with container. original_input : {ndarray, dataframe} Original input of function. estimator : estimator instance Estimator with to get the output configuration from. Returns ------- output : {ndarray, dataframe} If the output config is "default" or the estimator is not configured for wrapping return `data_to_wrap` unchanged. If the output config is "pandas", return `data_to_wrap` as a pandas DataFrame. rrr N)rr r )r"_auto_wrap_is_configuredrrget_feature_names_out)rroriginal_inputr output_configs r_wrap_data_with_containerr)ash2'vy99MW**2J92U2U* %!ngt44/   r#c@tfd}|S)z@Wrapper used by `_SetOutputMixin` to automatically wrap methods.c||g|Ri|}t|tr$t|d||g|ddRSt|||S)Nrr)rtupler))selfXargskwargsrfrs rwrappedz$_wrap_method_output..wrappedsqq24222622 lE * * *&,q/1dKKabb!  )q$GGGr#r)r1rr2s`` r_wrap_method_outputr3sA 1XX H H H H HX H Nr#cdt|dt}t|dod|vS)zReturn True if estimator is configured for auto-wrapping the transform method. `_SetOutputMixin` sets `_sklearn_auto_wrap_output_keys` to `set()` if auto wrapping is manually disabled. _sklearn_auto_wrap_output_keysr& transform)rsethasattr)rauto_wrap_output_keyss rr%r%s? $I/OQTQVQVWW 233 1 0 0r#cReZdZdZdfd ZeedddZxZS)_SetOutputMixina\Mixin that dynamically wraps methods to return container based on config. Currently `_SetOutputMixin` wraps `transform` and `fit_transform` and configures it based on `set_output` of the global configuration. `set_output` is only defined if `get_feature_names_out` is defined and `auto_wrap_output_keys` is the default value. r6c tjdi|t|ts|t d|t |_dSddd}t |_|D]m\}}t||r||vr|j |||j vr>tt|||}t|||ndS)Nz6auto_wrap_output_keys must be None or a tuple of keys.r6)r6 fit_transform)super__init_subclass__rr,rr7r5itemsr8add__dict__r3rsetattr)clsr9r0 method_to_keyrkeywrapped_method __class__s rrAz!_SetOutputMixin.__init_subclass__s$!!++F+++ ,e 4 4 W8M8UUVV V (14C . F%(  .1UU*(..00 1 1KFC3'' 36K+K+K  . 2 23 7 7 7S\))0f1E1EsKKN C 0 0 0 0 1 1r#NcP||St|dsi|_||jd<|S)aDSet output container. See :ref:`sphx_glr_auto_examples_miscellaneous_plot_set_output.py` for an example on how to use the API. Parameters ---------- transform : {"default", "pandas"}, default=None Configure output of `transform` and `fit_transform`. - `"default"`: Default output format of a transformer - `"pandas"`: DataFrame output - `None`: Transform configuration is unchanged Returns ------- self : estimator instance Estimator instance. Nrr6)r8r)r-r6s r set_outputz_SetOutputMixin.set_outputs<*  Kt566 -*,D '3<#K0 r#)r<) __name__ __module__ __qualname____doc__rAr r%rL __classcell__)rJs@rr;r;s|111111@\*++&*,+r#r;r<ct|dpt|do|du}|sdSt|dstd|d||S)aSafely call estimator.set_output and error if it not available. This is used by meta-estimators to set the output for child estimators. Parameters ---------- estimator : estimator instance Estimator instance. transform : {"default", "pandas"}, default=None Configure output of the following estimator's methods: - `"transform"` - `"fit_transform"` If `None`, this operation is a no-op. Returns ------- estimator : estimator instance Estimator instance. r6r>NrLzUnable to configure output for z' because `set_output` is not available.r<)r8rrL)rr6set_output_for_transforms r_safe_set_outputrTs0  ;'' " 9o . . " T ! $  9l + +  i         )  4 44r#)N) functoolsr scipy.sparserr_configr _available_ifr rr"r)r3r%r;rTr?r#rrZs!!!!!!"""""" ''''''  2D2D2D2D2Dj####D###L$   FFFFFFFFR.2&5&5&5&5&5&5&5r#