B &]\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/lib/python3.7/site-packages/scipy/odr/models.py_lin_fcn srcCs>t|jdt}t||f}|jd|jdf|_|S)N)nponesr float concatenateZravel)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$Zflat)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?)rZarray)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 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$)r.r/r0)r2r1r3Z extra_argsr4) rZasarrayr Zaranger rr&r(r')orderr%r5r6rrrr _s     Z Exponentialzy= B_0 + exp(B_1 * x)z$y=\beta_0 + e^{\beta_1 x}$)r2r1r3r4cCs||d|dS)Nrr r)rrrrr_unilinsr8cCst|jt|dS)Nr)rrr r)rrrrr _unilin_fjdsr9cCs(t|t|jtf}d|j|_|S)N)r)rrrr r)rr_retrrr _unilin_fjbs r;cCsdS)N)g?g?r)r!rrr _unilin_estsr<cCs |||d|d|dS)Nrr rr)rrrrr _quadraticsr=cCsd||d|dS)Nrrr r)rrrrr _quad_fjdsr>cCs.t|||t|jtf}d|j|_|S)N))rrrr r)rrr:rrr _quad_fjbs r@cCsdS)N)g?g?g?r)r!rrr _quad_estsrAzUnivariate 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__Z __future__rrrZnumpyrZscipy.odr.odrpackr__all__rrrr#r&r'r(r*r+r,r-rr rr8r9r;r<r=r>r@rArr rrrrsT     (