1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
|
#!/usr/bin/env python3
"""
External minimize: using lmfit minimizers for BornAgain fits.
Fit progress is plotted using lmfit iteration callback function.
"""
import bornagain as ba
from bornagain import ba_fitmonitor, nm
import lmfit
import model2_hexlattice as model
class LMFITPlotter:
"""
Adapts standard plotter for lmfit minimizer.
"""
def __init__(self, fit_objective, every_nth=10):
self.fit_objective = fit_objective
self.plotter_gisas = ba_fitmonitor.PlotterGISAS()
self.every_nth = every_nth
def __call__(self, P, i, resid):
if i % self.every_nth == 0:
self.plotter_gisas.plot(self.fit_objective)
if __name__ == '__main__':
data = model.fake_data()
fit_objective = ba.FitObjective()
fit_objective.addFitPair(model.get_simulation, data, 1)
fit_objective.initPrint(10)
P = lmfit.Parameters()
P.add('radius', value=7*nm, min=5*nm, max=8*nm)
P.add('length', value=10*nm, min=8*nm, max=14*nm)
result = lmfit.minimize(fit_objective.evaluate_residuals, P)
fit_objective.finalize(result)
result.params.pretty_print()
print(lmfit.fit_report(result))
|
