use more robust trf method instead lms

bmarechal
1 parent c81fde478c
Showing 1 changed file with 2 additions and 2 deletions Side-by-side Diff
fit_waist.py
@@ -23,8 +23,8 @@
 	Pmes = data[:,1]
  
 	'''optimization with non-linear least squares method'''
-	Ppopt, Pcov = curve_fit(P, xmes, Pmes)
-	data_waist.append([int(f[-7:-4]), abs(Ppopt[3])])
+	Ppopt, Pcov = curve_fit(P, xmes, Pmes, method = 'trf')
+	data_waist.append([int(f[-7:-4]), Ppopt[3]])
  
 	'''plot'''
 	p[0].plot(xmes, Pmes, 'o')
...	...	@@ -23,8 +23,8 @@
23	23	Pmes = data[:,1]
24	24
25	25	'''optimization with non-linear least squares method'''
26		- Ppopt, Pcov = curve_fit(P, xmes, Pmes)
27		- data_waist.append([int(f[-7:-4]), abs(Ppopt[3])])
	26	+ Ppopt, Pcov = curve_fit(P, xmes, Pmes, method = 'trf')
	27	+ data_waist.append([int(f[-7:-4]), Ppopt[3]])
28	28
29	29	'''plot'''
30	30	p[0].plot(xmes, Pmes, 'o')