3 6^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)Bxabr1/tmp/pip-build-vw4w4j08/scipy/scipy/odr/models.py_lin_fcn srcCs>tj|jdt}tj||jf}|jd|jdf|_|S)Nr rr)nponesr float concatenateZravel)rrrresrrr_lin_fjbsrcCs:|dd}tj||jdf|jddd}|j|_|S)Nr r)r rr)rrepeatr )rrrrrr_lin_fjds "rcCs4t|jjdkr|jjd}nd}tj|dftS)Nrr )lenrr rrr)datamrrr_lin_est!sr#cCsD|d|dd}}|jddf|_|tj|tj||ddS)Nrr )r )r rrpower)rrpowersrrrrr _poly_fcn.sr&cCs@tjtj|jdttj||jf}|jd|jdf|_|S)Nr rrr)rrrr rr$Zflat)rrr%rrrr _poly_fjacb5s r'cCsB|dd}|jddf|_||}tj|tj||dddS)Nr r)r )r rrr$)rrr%rrrr _poly_fjacd<s r(cCs|dtj|d|S)Nrr )rexp)rrrrr_exp_fcnEsr*cCs|dtj|d|S)Nr )rr))rrrrr_exp_fjdIsr+cCsBtjtj|jdt|tj|d|f}d|jdf|_|S)Nr rrr)rrrr rr))rrrrrr_exp_fjbMs.r,cCstjddgS)Ng?)rarray)r!rrr_exp_estSsr.zArbitrary-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 Csxtj|}|jfkr$tjd|d}t|df|_t|d}|fdd}tttt||fdd|dd|dddS) 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. r cSstj|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$)r/r0r1)r3r2r4 extra_argsr5) rZasarrayr Zaranger rr&r(r')orderr%r6r7rrrr _s     Z Exponentialzy= B_0 + exp(B_1 * x)z$y=\beta_0 + e^{\beta_1 x}$)r3r2r4r5cCs||d|dS)Nrr r)rrrrr_unilinsr:cCstj|jt|dS)Nr)rrr r)rrrrr _unilin_fjdsr;cCs(tj|tj|jtf}d|j|_|S)Nr)r)rrrr r)rr_retrrr _unilin_fjbs r=cCsdS)N?)r>r>r)r!rrr _unilin_estsr?cCs |||d|d|dS)Nrr rr)rrrrr _quadraticsr@cCsd||d|dS)Nrrr r)rrrrr _quad_fjdsrAcCs.tj|||tj|jtf}d|j|_|S)N)rB)rrrr r)rrr<rrr _quad_fjbs rCcCsdS)N?)rDrDrDr)r!rrr _quad_estsrEzUnivariate 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__rrrZnumpyrZscipy.odr.odrpackr__all__rrrr#r&r'r(r*r+r,r.rr rr:r;r=r?r@rArCrErr rrrrsT     (