B 7 [7 @sHdZddlmZmZmZddlZddlmZddddd d gZ d d Z d dZ ddZ ddZ ddZddZddZddZddZddZdd Zee e e e d!d"d#d$d%Zd&d Zeeeeed'd(d)d$d*Zd+d,Zd-d.Zd/d0Zd1d2Zd3d4Zd5d6Zd7d8Zd9d:Zeeeeed;dd?d@d$d*Z!dS)AzI Collection of Model instances for use with the odrpack fitting package. )divisionprint_functionabsolute_importN)Modelr exponential multilinear unilinear quadratic polynomialcCs:|d|dd}}|jddf|_|||jddS)Nr)axis)shapesum)Bxabr\/home/fristb/BIRCH/lib-linux-x86_64/python/lib64/python3.7/site-packages/scipy/odr/models.py_lin_fcn srcCs>t|jdt}t||f}|jd|jdf|_|S)N)nponesr float concatenateravel)rrrresrrr_lin_fjbsrcCs:|dd}tj||jdf|jddd}|j|_|S)Nr rr)r )rrepeatr )rrrrrr_lin_fjds "rcCs4t|jjdkr|jjd}nd}t|dftS)Nrr )lenrr rrr)datamrrr_lin_est!sr$cCsD|d|dd}}|jddf|_|tj|t||ddS)Nrr )r )r rrpower)rrpowersrrrrr _poly_fcn.sr'cCs@tt|jdtt||jf}|jd|jdf|_|S)Nr)rrrr rr%flat)rrr&rrrr _poly_fjacb5s r)cCsB|dd}|jddf|_||}tj|t||dddS)Nr r)r )r rrr%)rrr&rrrr _poly_fjacd<s r*cCs|dt|d|S)Nrr )rexp)rrrrr_exp_fcnEsr,cCs|dt|d|S)Nr )rr+)rrrrr_exp_fjdIsr-cCsBtt|jdt|t|d|f}d|jdf|_|S)Nrr r )rrrr rr+)rrrrrr_exp_fjbMs.r.cCstddgS)Ng?)rarray)r"rrr_exp_estSsr0zArbitrary-dimensional Linearz y = B_0 + Sum[i=1..m, B_i * x_i]z&$y=\beta_0 + \sum_{i=1}^m \beta_i x_i$)nameequTeXequ)fjacbfjacdestimatemetac Csxt|}|jdkr$td|d}t|df|_t|d}|fdd}tttt||fdd|dd|ddd S) a Factory function for a general polynomial model. Parameters ---------- order : int or sequence If an integer, it becomes the order of the polynomial to fit. If a sequence of numbers, then these are the explicit powers in the polynomial. A constant term (power 0) is always included, so don't include 0. Thus, polynomial(n) is equivalent to polynomial(range(1, n+1)). Returns ------- polynomial : Model instance Model instance. rr cSst|ftS)N)rrr)r"len_betarrr _poly_est{szpolynomial.._poly_estzSorta-general Polynomialz$y = B_0 + Sum[i=1..%s, B_i * (x**i)]z)$y=\beta_0 + \sum_{i=1}^{%s} \beta_i x^i$)r1r2r3)r5r4r6 extra_argsr7) rasarrayr aranger!rr'r*r))orderr&r8r9rrrr _s     Z Exponentialzy= B_0 + exp(B_1 * x)z$y=\beta_0 + e^{\beta_1 x}$)r5r4r6r7cCs||d|dS)Nrr r)rrrrr_unilinsr>cCst|jt|dS)Nr)rrr r)rrrrr _unilin_fjdsr?cCs(t|t|jtf}d|j|_|S)N)r )rrrr r)rr_retrrr _unilin_fjbs rAcCsdS)N)g?g?r)r"rrr _unilin_estsrBcCs |||d|d|dS)Nrr r r)rrrrr _quadraticsrCcCsd||d|dS)Nr rr r)rrrrr _quad_fjdsrDcCs.t|||t|jtf}d|j|_|S)N))rrrr r)rrr@rrr _quad_fjbs rFcCsdS)N)g?g?g?r)r"rrr _quad_estsrGzUnivariate Linearzy = B_0 * x + B_1z$y = \beta_0 x + \beta_1$Z Quadraticzy = B_0*x**2 + B_1*x + B_2z&$y = \beta_0 x^2 + \beta_1 x + \beta_2)"__doc__ __future__rrrnumpyrZscipy.odr.odrpackr__all__rrrr$r'r)r*r,r-r.r0rr rr>r?rArBrCrDrFrGrr rrrrsT     (