B 0dTH@sdZddlZddlZddlmZddlmZddlmZddl m Z Gd d d e Z Gd d d e Z d dZGdddejZdddZdS)ztThis module contains the _EstimatorPrettyPrinter class used in BaseEstimator.__repr__ for pretty-printing estimatorsN) OrderedDict) BaseEstimator) get_config) is_scalar_nancs eZdZdZfddZZS) KeyValTuplez@Dummy class for correctly rendering key-value tuples from dicts.cs tS)N)super__repr__)self) __class__G/var/www/html/venv/lib/python3.7/site-packages/sklearn/utils/_pprint.pyr NszKeyValTuple.__repr__)__name__ __module__ __qualname____doc__r __classcell__r r )r rrKsrc@seZdZdZdS)KeyValTupleParamzEDummy class for correctly rendering key-value tuples from parameters.N)rrrrr r r rrSsrcs\|jdd}t|jd|j}t|jddDfddfdd|DS) zgReturn dict (param_name: value) of parameters that were given to estimator with non-default values.F)deepZdeprecated_originalcSsi|]\}}|j|qSr )default).0nameparamr r r `sz#_changed_params..csl|kr dS|tjkrdSt|tr<|j|jkr.has_changedcs i|]\}}||r||qSr r )rrr )r"r rrqs) get_paramsgetattr__init__r signature parametersitems)Z estimatorparamsZ init_funcr )r"r!r_changed_paramsYs    r*cseZdZdZdddddfdd 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||_|jr(d|_td|_||_dS)N)r-rZprint_changed_only)r r%_indent_at_name_indent_per_levelr _changed_onlyr/)r indentwidthdepthstreamr-r.r/)r r rr%s  z _EstimatorPrettyPrinter.__init__cCst|||||jdS)N) changed_only) _safe_reprr2)r objectcontext maxlevelslevelr r rformatsz_EstimatorPrettyPrinter.formatcCs||jjd|jr(|t|jj7}|jr8t|}n |jdd}tddt | D}| | |||d|||ddS)N(F)rcss|]\}}||fVqdS)Nr )rrvalr r r sz<_EstimatorPrettyPrinter._pprint_estimator..r)) writer rr0lenr2r*r#rsortedr(_format_params)r r9r6r3 allowancer:r<r)r r r_pprint_estimators  z)_EstimatorPrettyPrinter._pprint_estimatorc Cs|j||||||ddS)NT)is_dict)_format_params_or_dict_items)r r(r6r3rFr:r<r r r_format_dict_itemssz*_EstimatorPrettyPrinter._format_dict_itemsc Cs|j||||||ddS)NF)rH)rI)r r(r6r3rFr:r<r r rrEsz&_EstimatorPrettyPrinter._format_paramscCs|j}||j7}dd|} d} |j|d} } t|} y t| }Wntk rZdSXd}d}x>|s||jkr|dP|d7}|}y t| }Wn(tk rd }| |8} | |8} YnX|jrf|\}}||||}||||}|s| d }|r d nd }|||}t |d }| |kr>| } | r>| } | |krf| |8} || d} ||qh|| | } |r|t nt }| |||||r|nd||qhWdS)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. z,  rNFrz, ...T'z: =rz, )rBr1_widthiternext StopIterationr/_compact_reprstriprCrr_format)r r9r6r3rFr:r<rHrBdelimnldelimr4 max_widthitnext_entlastn_itemsentrr kreprvreprmiddlerepwclass_r r rrIs`            z4_EstimatorPrettyPrinter._format_params_or_dict_itemscCsr|j}||j7}|jdkr,||jdddd|}d} |j|d} } t|} y t| } Wntk rvdSXd}d}x|sl||jkr|dP|d7}| }y t| } Wn(tk rd }| |8} | |8} YnX|jr@||||}t |d }| |kr| } | r|} | |kr@| |8} || d } ||q|| |} | ||||r`|nd||qWdS) 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. rrKz, rLNFrz, ...Trz, ) rBr1rOrPrQrRr/rSrTrCrV)r r(r6r3rFr:r<rBrWrXr4rYrZr[r\r]r^rbrcr r r _format_itemssT         z%_EstimatorPrettyPrinter._format_itemsc Csp|\}}||||} t|tr0| d} d} nd} || || ||||t| t| |||dS)z=Pretty printing for key-value tuples from dict or parameters.rMrNz: N)rTrrrUrBrVrC) r r9r6r3rFr:r<rr rbrar r r_pprint_key_val_tupleKs    z-_EstimatorPrettyPrinter._pprint_key_val_tuple)rr,NN)rrrrr%r=rGrJrErIrerfpprint PrettyPrinter _dispatchcopyrr rrr r )r rr+ts$1 ?0  r+FcCst|}|tjkr t|ddfSt|dd}t|trH|tjkrH|sLdSt|}|rn||krndd||kfS||krt |ddfSd||<d}d} g} | j } |d7}t } t | tjd} xn| D]f\}}| |||||d \}}}| |||||d \}}}| d ||f|o|o|}|s&|rd} qW||=d d | || fSt|tr`|tjksxt|trz|tjkrzt|tr|sd Sd}n"t|dkrd}n|sdSd}t|}|r||kr|dd||kfS||krt |ddfSd||<d}d} g} | j } |d7}xB|D]:}t |||||d \}}}| ||sPd}|r"d} q"W||=|d | || fSt|trt|}|r||krdd||kfS||krt |ddfSd||<d}d} |rt|}n |jdd}g} | j } |d7}t } t | tjd} xx| D]p\}}| |||||d \}}}| |||||d \}}}| d|d|f|o||o||}|s|r d} q W||=d|jd | f|| fSt|}||o|d dfS)zMSame as the builtin _safe_repr, with added support for Estimator objects.TFr N)z{}TFz{...}r)key)r7z%s: %sz{%s}z, )z[]TFz[%s]z(%s,))z()TFz(%s)z...)rz%s=%srMz%s(%s)<)typerg_builtin_scalarsrr$ issubclassdictr id _recursionappendr8rDr( _safe_tuplejoinlisttuplerCrr*r#rUr startswith)r9r:r;r<r7typrobjidreadable recursive componentsrssafereprr(rr r_ kreadablekrecurr` vreadablevrecurr=oorepr oreadableorecurr)rbr r rr8cs              r8)F)rrrg collectionsrbaser_configrrLrrwrrr*rhr+r8r r r rs@    p