J/PhddlmZddlZddlmZmZmZmZm Z m Z m Z gdZ dZ GddeZe eZGdd eZe eZeZdS) )wrapsN)atleast_2d_column_defaultasarray_or_pandaspandas_friendly_reshapewide_dtype_forsafe_issubdtype no_picklingassert_no_pickling)stateful_transformcenter standardizescalecJtfd}|_|S)zCreate a stateful transform callable object from a class that fulfills the :ref:`stateful transform protocol `. ct}|j|i|||j|i|SN)memorize_chunkmemorize_finish transform)argskwargsrclass_s K/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/patsy/state.pystateful_transform_wrapperz6stateful_transform..stateful_transform_wrapper5sPFHH   $1&111!!###"y"D3F333)r__patsy_stateful_transform__)rrs` rr r 0s@  6]]4444]4 ?E; %%rc.eZdZdZdZdZdZdZeZ dS)Centerzcenter(x) A stateful transform that centers input data, i.e., subtracts the mean. If input has multiple columns, centers each column separately. Equivalent to ``standardize(x, rescale=False)`` c"d|_d|_dSNr)_sum_countselfs r__init__zCenter.__init__fs  rct|}|xj|jdz c_tj|dt |}|j ||_dS|xj|z c_dS)Nrdtype)rr!shapenpsumrr )r#x this_totals rrzCenter.memorize_chunkjsj %a ( ( qwqz! VAqq(9(9::: 9 "DIII II #IIIIrcdSrr"s rrzCenter.memorize_finisht rc<t|}tj|}t|jtjrt }n|j}tj|j|jz |}t|d|z }t||j S)Nr&Tpreserve_pandas) rr)asarrayrr'integerfloatr r!rrr()r#r+x_arrdtmean_valcentereds rrzCenter.transformws a  1  5; 3 3 BBB:di$+5R@@@,QEEEP&x999rN __name__ __module__ __qualname____doc__r$rrrr __getstate__r.rrrr\s[$$$    : : :LLLrrc2eZdZdZdZd dZdZd dZeZ dS) StandardizeaHstandardize(x, center=True, rescale=True, ddof=0) A stateful transform that standardizes input data, i.e. it subtracts the mean and divides by the sample standard deviation. Either centering or rescaling or both can be disabled by use of keyword arguments. The `ddof` argument controls the delta degrees of freedom when computing the standard deviation (cf. :func:`numpy.std`). The default of ``ddof=0`` produces the maximum likelihood estimate; use ``ddof=1`` if you prefer the square root of the unbiased estimate of the variance. If input has multiple columns, standardizes each column separately. .. note:: This function computes the mean and standard deviation using a memory-efficient online algorithm, making it suitable for use with large incrementally processed data-sets. c0d|_d|_d|_dSr) current_n current_mean current_M2r"s rr$zStandardize.__init__s rTrct|}|jftj|jdt ||_tj|jdt ||_t|jdD]c}|xjdz c_||ddf|jz }|xj||jz z c_|xj|||ddf|jz zz c_ddS)Nr&r) rrDr)zerosr(rrErangerC)r#r+r rescaleddofideltas rrzStandardize.memorize_chunks %a ( (   $ ">!;L;L M M MD  hqwqz9J9JKKKDOqwqz"" E EA NNa NNadGd//E   !7 7   OOu!QQQ$$2C(CD DOOO  E ErcdSrr.r"s rrzStandardize.memorize_finishr/rct|dt}t|d}|r ||jz}|r'|t j|j|j|z z z}t||j S)NT)copyr'r1) rr5rrDr)sqrtrErCrr()r#r+r rJrKx_2ds rrzStandardize.transforms| ad% 8 8 8(DAAA  & D% %D  G BGDOt~/DEFF FD&tQW555rN)TTrr:r.rrrArAsi$ E E E E    6 6 6 6LLLrrA) functoolsrnumpyr) patsy.utilrrrrrr r __all__r objectrr rAr rr.rrrXs#6    & & &X)))))V)))X  F # # 44444&444n! -- r