o ?HhJH@sdZddlZddlZddlmZddlmZddlmZGdd d e Z Gd d d e Z d d Z Gdddej ZdddZdS)ztThis module contains the _EstimatorPrettyPrinter class used in BaseEstimator.__repr__ for pretty-printing estimatorsN) get_config) BaseEstimator) is_scalar_nancs eZdZdZfddZZS) KeyValTuplez@Dummy class for correctly rendering key-value tuples from dicts.cs tS)N)super__repr__)self __class__U/home/air/sanwanet/gpt-api/venv/lib/python3.10/site-packages/sklearn/utils/_pprint.pyr Ps zKeyValTuple.__repr__)__name__ __module__ __qualname____doc__r __classcell__r r r rrMsrc@seZdZdZdS)KeyValTupleParamzEDummy class for correctly rendering key-value tuples from parameters.N)rrrrr r r rrUsrcs\|jdd}t|jd|j}t|jddDfddfdd|DS) zgReturn dict (param_name: value) of parameters that were given to estimator with non-default values.Fdeepdeprecated_originalcSsi|]\}}||jqSr )default).0nameparamr r r bsz#_changed_params..csl|vrdS|tjkrdSt|tr|j|jkrdSt|t|kr4t|r2t|s4dSdS)NTF)inspect_empty isinstancerr reprr)kv) init_paramsr r has_changedds z$_changed_params..has_changedcs i|] \}}||r||qSr r )rr!r")r$r rrss ) get_paramsgetattr__init__r signature parametersitems) estimatorparams init_funcr )r$r#r_changed_params[s   r.cseZdZdZ    dddddfdd Zd d Zd d ZddZddZddZ ddZ ddZ e j jZeeej<e eej<ZS)_EstimatorPrettyPrintera Pretty Printer class for estimator objects. This extends the pprint.PrettyPrinter class, because: - we need estimators to be printed with their parameters, e.g. Estimator(param1=value1, ...) which is not supported by default. - the 'compact' parameter of PrettyPrinter is ignored for dicts, which may lead to very long representations that we want to avoid. Quick overview of pprint.PrettyPrinter (see also https://stackoverflow.com/questions/49565047/pprint-with-hex-numbers): - the entry point is the _format() method which calls format() (overridden here) - format() directly calls _safe_repr() for a first try at rendering the object - _safe_repr formats the whole object recursively, only calling itself, not caring about line length or anything - back to _format(), if the output string is too long, _format() then calls the appropriate _pprint_TYPE() method (e.g. _pprint_list()) depending on the type of the object. This where the line length and the compact parameters are taken into account. - those _pprint_TYPE() methods will internally use the format() method for rendering the nested objects of an object (e.g. the elements of a list) In the end, everything has to be implemented twice: in _safe_repr and in the custom _pprint_TYPE methods. Unfortunately PrettyPrinter is really not straightforward to extend (especially when we want a compact output), so the code is a bit convoluted. This class overrides: - format() to support the changed_only parameter - _safe_repr to support printing of estimators (for when they fit on a single line) - _format_dict_items so that dict are correctly 'compacted' - _format_items so that ellipsis is used on long lists and tuples When estimators cannot be printed on a single line, the builtin _format() will call _pprint_estimator() because it was registered to do so (see _dispatch[BaseEstimator.__repr__] = _pprint_estimator). both _format_dict_items() and _pprint_estimator() use the _format_params_or_dict_items() method that will format parameters and key-value pairs respecting the compact parameter. This method needs another subroutine _pprint_key_val_tuple() used when a parameter or a key-value pair is too long to fit on a single line. This subroutine is called in _format() and is registered as well in the _dispatch dict (just like _pprint_estimator). We had to create the two classes KeyValTuple and KeyValTupleParam for this. rPNFT)compactindent_at_namen_max_elements_to_showcs>tj|||||d||_|jrd|_td|_||_dS)N)r1rprint_changed_only)rr'_indent_at_name_indent_per_levelr _changed_onlyr3)r indentwidthdepthstreamr1r2r3r r rr's    z _EstimatorPrettyPrinter.__init__cCst|||||jdS)N changed_only) _safe_reprr7)r objectcontext maxlevelslevelr r rformats z_EstimatorPrettyPrinter.formatcCsr||jjd|jr|t|jj7}|jrt|}n|jdd}|t | |||d|||ddS)N(Frr)) writer rr5lenr7r.r%_format_paramssortedr*)r r?r;r8 allowancer@rBr,r r r_pprint_estimators  z)_EstimatorPrettyPrinter._pprint_estimatorc C|j||||||ddS)NTis_dict_format_params_or_dict_itemsr r*r;r8rJr@rBr r r_format_dict_itemsz*_EstimatorPrettyPrinter._format_dict_itemsc CrL)NFrMrOrQr r rrHrSz&_EstimatorPrettyPrinter._format_paramscCs|j}||j7}dd|} d} |j|d} } t|} zt| }Wn ty-YdSwd}d}|s||jkr?|ddS|d7}|}zt| }Wnty^d }| |8} | |8} Ynw|jr|\}}||||}||||}|s{| d }|rd nd }|||}t |d }| |kr| } | r| } | |kr| |8} || d} ||q2|| | } |rt nt }| |||||r|nd|||r4dSdS)a-Format dict items or parameters respecting the compact=True parameter. For some reason, the builtin rendering of dict items doesn't respect compact=True and will use one line per key-value if all cannot fit in a single line. Dict items will be rendered as <'key': value> while params will be rendered as . The implementation is mostly copy/pasting from the builtin _format_items(). This also adds ellipsis if the number of items is greater than self.n_max_elements_to_show. ,  rNFr, ...T': =r, )rFr6_widthiternext StopIterationr3_compact_reprstriprGrr_format)r r?r;r8rJr@rBrNrFdelimnldelimr9 max_widthitnext_entlastn_itemsentr!r"kreprvreprmiddlerepwclass_r r rrPsh              z4_EstimatorPrettyPrinter._format_params_or_dict_itemscCsh|j}||j7}|jdkr||jdddd|}d} |j|d} } t|} zt| } Wn ty;YdSwd}d}|s||jkrM|ddS|d7}| }zt| } Wntyld }| |8} | |8} Ynw|jr||||}t |d }| |kr| } | r|} | |kr| |8} || d } ||q@|| |} | ||||r|nd|||rBdSdS) zFormat the items of an iterable (list, tuple...). Same as the built-in _format_items, with support for ellipsis if the number of elements is greater than self.n_max_elements_to_show. rrUrTrVNFrrWTrr[) rFr6r\r]r^r_r3r`rarGrc)r r*r;r8rJr@rBrFrdrer9rfrgrhrirjrkrorpr r r _format_itemssZ           z%_EstimatorPrettyPrinter._format_itemsc Csp|\}}||||} t|tr| d} d} nd} || || ||||t| t| |||dS)z=Pretty printing for key-value tuples from dict or parameters.rXrZrYN)rarrrbrFrcrG) r r?r;r8rJr@rBr!r"rornr r r_pprint_key_val_tupleKs    z-_EstimatorPrettyPrinter._pprint_key_val_tuple)rr0NN)rrrrr'rCrKrRrHrPrrrspprint PrettyPrinter _dispatchcopyrr rrr r r rr/vs*4?0  r/FcCst|}|tjvrt|ddfSt|dd}t|tr|tjur|s$dSt|}|r5||kr5dd||vfS||vrAt |ddfSd||<d}d} g} | j } |d7}t } t | tjd} | D]0\}}| |||||d \}}}| |||||d \}}}| d ||f|o|o|}|s|rd} q_||=d d | || fSt|tr|tjust|tr+|tjur+t|tr|sd Sd}nt|dkrd}n|sdSd}t|}|r||kr|dd||vfS||vrt |ddfSd||<d}d} g} | j } |d7}|D]}t |||||d \}}}| ||sd}|rd} q||=|d | || fSt|trt|}|rH||krH|jdd||vfS||vrUt |ddfSd||<d}d} |ret|}n|jdd}g} | j } |d7}t } t | tjd} | D]8\}}| |||||d \}}}| |||||d \}}}| d|d|f|o|o|}|s|rd} q||=d|jd | f|| fSt|}||o|d dfS)zMSame as the builtin _safe_repr, with added support for Estimator objects.TFr N)z{}TFz{...}r)keyr<z%s: %sz{%s}r[)z[]TFz[%s]z(%s,))z()TFz(%s)z...z(...)rz%s=%srXz%s(%s)<)typert_builtin_scalarsr r& issubclassdictr id _recursionappendr>rIr* _safe_tuplejoinlisttuplerGrrr.r%rb startswith)r?r@rArBr=typrobjidreadable recursive componentsrsafereprr*r!r"rl kreadablekrecurrm vreadablevrecurrCoorepr oreadableorecurr,ror r rr>cs                         r>)F)rrrt_configrbaser_missingrrrrr.rur/r>r r r rsD   n