def myopt2(p, x, z, model, skyx, sky, skymodel, r, o): par = special_functions.unpack_coeff(p) coeff, ier = optimize.leastsq(skyarcfit, par, (x, z, model, skyx, sky, skymodel, p, r, o), maxfev=100000) par = special_functions.build_coeff(coeff, p) return par
def myoptimize(p, x, z, model, model2=None): par = special_functions.unpack_coeff(p) mask = ndimage.maximum_filter(z, 3) coeff, ier = optimize.leastsq(arcfitfunc, par, (x, z, model, p, model2, mask), maxfev=100000) par = special_functions.build_coeff(coeff, p) return par
def skyopt(p, x, data, model): par = special_functions.unpack_coeff(p).tolist() wave = special_functions.genfunc(x, 0, p).astype(scipy.float64) sky = interpolate.splev(wave, model).astype(scipy.float64) ratio = scipy.median(data) / scipy.median(sky) offset = 0. par.append(ratio) par.append(offset) coeff, ier = optimize.leastsq(skyfit, par, (x, data, model, p), maxfev=100000) return special_functions.build_coeff(coeff, p), coeff[-2], coeff[-1]
def myoptimize(p,x,z,scale,model,model2=None): par = special_functions.unpack_coeff(p) coeff,ier = optimize.leastsq(arcfitfunc,par,(x,z,scale,model,p,model2),maxfev=100000) par = special_functions.build_coeff(coeff,p) return par
def arcfitfunc2(p,x,y,z,model): par = special_functions.unpack_coeff(p) coeff,ier = optimize.leastsq(doarcfitfunc,par,(x,y,z,model,p),maxfev=100000) return special_functions.build_coeff(coeff,p)