B vd)@sdZddlmZddlZddlZddlZddlmZddl Z ddl Z ddl m Z ddlZddlmZmZddlmZeejZee jZeedkrdd l mZn dd lmZydd lmZmZWn$ek rdd lmZmZYnXd d Zd dZ ddZ!Gddde j"j#Z$ddZ%ddZ&GdddZ'd*ddZ(ddd d!Z)eed"kr^e)Z*n dd#lm*Z*d$d%Z+d+d&d'Z,ej,je,_d(d)Z-ej-je-_dS),zCompatibility fixes for older version of python, numpy and scipy If you add content to this file, please give the version of the package at which the fix is no longer needed. )update_wrapperN)lsqr)config_context get_config)parsez1.4)lobpcg)line_search_wolfe2line_search_wolfe1cCs||kS)N)Xr r `/work/yifan.wang/ringdown/master-ringdown-env/lib/python3.7/site-packages/sklearn/utils/fixes.py_object_dtype_isnan-srcCs&ttdkst|sddiSiSdS)zReturns the copy=False parameter for {ndarray, csr_matrix, csc_matrix}.astype when possible, otherwise don't specify z1.1copyFN) sp_version parse_versionspissparse)r r r r _astype_copy_false2srcKsddl}t|jtdkr|St|ddh}|rNtdt||jfi}d|kr|d}|dkrvtd|d d dd||d <d|kr|d}|d krtd ||dkrd |d <|S)aSet joblib.Parallel arguments in a compatible way for 0.11 and 0.12+ For joblib 0.11 this maps both ``prefer`` and ``require`` parameters to a specific ``backend``. Parameters ---------- prefer : str in {'processes', 'threads'} or None Soft hint to choose the default backend if no specific backend was selected with the parallel_backend context manager. require : 'sharedmem' or None Hard condstraint to select the backend. If set to 'sharedmem', the selected backend will be single-host and thread-based even if the user asked for a non-thread based backend with parallel_backend. See joblib.Parallel documentation for more details rNz0.12preferrequirez%unhandled arguments %s with joblib %s)threadsZ processesNzprefer=%s is not supported threadingmultiprocessingbackend)N sharedmemzrequire=%s is not supportedr) joblibr __version__setkeys differenceNotImplementedErrorlist ValueError)kwargsr extra_argsargsrrr r r _joblib_parallel_args=s0   r'c@seZdZdZdS) loguniformawA class supporting log-uniform random variables. Parameters ---------- low : float The minimum value high : float The maximum value Methods ------- rvs(self, size=None, random_state=None) Generate log-uniform random variables The most useful method for Scikit-learn usage is highlighted here. For a full list, see `scipy.stats.reciprocal `_. This list includes all functions of ``scipy.stats`` continuous distributions such as ``pdf``. Notes ----- This class generates values between ``low`` and ``high`` or low <= loguniform(low, high).rvs() <= high The logarithmic probability density function (PDF) is uniform. When ``x`` is a uniformly distributed random variable between 0 and 1, ``10**x`` are random variables that are equally likely to be returned. This class is an alias to ``scipy.stats.reciprocal``, which uses the reciprocal distribution: https://en.wikipedia.org/wiki/Reciprocal_distribution Examples -------- >>> from sklearn.utils.fixes import loguniform >>> rv = loguniform(1e-3, 1e1) >>> rvs = rv.rvs(random_state=42, size=1000) >>> rvs.min() # doctest: +SKIP 0.0010435856341129003 >>> rvs.max() # doctest: +SKIP 9.97403052786026 N)__name__ __module__ __qualname____doc__r r r r r(qs.r(c Csttdkrtj|||dS|dkr,|}t|jtjsDtd|j |j krXt dd|j }t t |dgt t |d|j }g}xbt ||jD]R\}}|dkr||q|d|d||dd}|t||qWt|}||SdS) zQImplements a simplified version of np.take_along_axis if numpy version < 1.15z1.15)arrindicesaxisNz"`indices` must be an integer arrayz;`indices` and `arr` must have the same number of dimensions)r0)) np_versionrnpZtake_along_axisflattenZ issubdtypedtypeZintp IndexErrorndimr#r"rangezipshapeappendZarangeZreshapetuple) r-r.r/Z shape_onesZ dest_dimsZ fancy_indexdimnZ ind_shaper r r _take_along_axiss&   &  r?cstfdd}|S)z6Decorator used to capture the arguments of a function.cst||fS)N) _FuncWrapper)r&r$)functionr r delayed_functionsz!delayed..delayed_function) functoolswraps)rArBr )rAr delayedsrEc@s eZdZdZddZddZdS)r@z< "Load the global configuration before calling the function.cCs||_t|_t||jdS)N)rArconfigr)selfrAr r r __init__sz_FuncWrapper.__init__c Os$tf|j|j||SQRXdS)N)rrFrA)rGr&r$r r r __call__sz_FuncWrapper.__call__N)r)r*r+r,rHrIr r r r r@sr@2TFc CsZttdkr>t|d}t|d}t||t|}|dkrH|}|jdkr`|jkrznntj||||||dS|jdks|jdks|j|jkrt d|jd}tj ||f|d} tj |tjd} xBt |D]6} tj|| || ||d |d\| dd| f<| | <qW|dkr*t | d|} |r8| | fS| Sntj|||||||d SdS) a1Implements a simplified linspace function as of numpy version >= 1.16. As of numpy 1.16, the arguments start and stop can be array-like and there is an optional argument `axis`. For simplicity, we only allow 1d array-like to be passed to start and stop. See: https://github.com/numpy/numpy/pull/12388 and numpy 1.16 release notes about start and stop arrays for linspace logspace and geomspace. Returns ------- out : ndarray of shape (num, n_start) or (num,) The output array with `n_start=start.shape[0]` columns. z1.16g?Nr)startstopnumendpointretstepr5r0z3start and stop must be 1d array-like of same shape.)r5T)rKrLrMrNrOr5r/) r2rr3Z asanyarrayZ result_typefloatr7linspacer:r#emptyr8Zmoveaxis) rKrLrMrNrOr5r/dtZn_startoutstepir r r rQsP    rQZlinear)methodcKstj||fd|i|S)N interpolation)r3 percentile)aqrWr$r r r _percentilesr\z1.22)rYcCs(ttdsdSttds"tt_tjS)NThreadpoolController_sklearn_threadpool_controller)hasattr threadpoolctlsklearnr]r^r r r r _get_threadpool_controller+s    rbcCs.t}|dk r|j||dStj||dSdS)N)limitsuser_api)rblimitr`threadpool_limits)rcrd controllerr r r rf5srfcCs"t}|dk r|StSdS)N)rbinfor`threadpool_info)rgr r r ri@sri)rJTFNr)NN).r,rCrranumpyr3Z scipy.sparsesparserZscipyZ scipy.statsZscipy.sparse.linalgrZ sparse_lsqrr`_configrrZexternals._packaging.versionrrrr2rrZexternals._lobpcgZscipy.optimize._linesearchr r ImportErrorZscipy.optimize.linesearchrrr'statsZ reciprocalr(r?rEr@rQr\rYrbrfrir r r r sJ         41"  B