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
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
|
#!/usr/bin/env python3
"""
Fitting example: running one fit after another using different minimizers.
During the first fit Genetic minimizer will be used to scan large parameter
space and roughly find local minimum. The second Minuit2 minimizer will continue
after that to find precise minimum location.
"""
import numpy as np
from matplotlib import pyplot as plt
import bornagain as ba
from bornagain import ba_fitmonitor, deg, angstrom, nm
def get_sample(params):
"""
Returns a sample with uncorrelated cylinders and pyramids on a substrate.
"""
radius = params["radius"]
height = params["height"]
m_vacuum = ba.RefractiveMaterial("Vacuum", 0, 0)
m_substrate = ba.RefractiveMaterial("Substrate", 6e-6, 2e-8)
m_particle = ba.RefractiveMaterial("Particle", 6e-4, 2e-8)
cylinder_ff = ba.Cylinder(radius, height)
cylinder = ba.Particle(m_particle, cylinder_ff)
layout = ba.ParticleLayout()
layout.addParticle(cylinder)
vacuum_layer = ba.Layer(m_vacuum)
vacuum_layer.addLayout(layout)
substrate_layer = ba.Layer(m_substrate, 0)
sample = ba.MultiLayer()
sample.addLayer(vacuum_layer)
sample.addLayer(substrate_layer)
return sample
def get_simulation(params):
"""
Returns a GISAXS simulation with beam and detector defined.
"""
beam = ba.Beam(1e8, 1*angstrom, 0.2*deg)
n = 100 # bp.simargs['n']
detector = ba.SphericalDetector(n, 0, 2*deg, n, 0, 2*deg)
return ba.ScatteringSimulation(beam, get_sample(params), detector)
def create_real_data():
"""
Generating "real" data by adding noise to the simulated data.
"""
params = {'radius': 5*nm, 'height': 5*nm}
simulation = get_simulation(params)
result = simulation.simulate()
# retrieving simulated data in the form of numpy array
real_data = result.array()
# spoiling simulated data with the noise to produce "real" data
noise_factor = 0.1
noisy = np.random.normal(real_data, noise_factor*np.sqrt(real_data))
noisy[noisy < 0.1] = 0.1
return noisy
def run_fitting():
"""
main function to run fitting
"""
real_data = create_real_data()
fit_objective = ba.FitObjective()
fit_objective.addSimulationAndData(get_simulation, real_data, 1)
fit_objective.initPrint(10)
observer = ba_fitmonitor.PlotterGISAS()
fit_objective.initPlot(10, observer)
"""
Setting fitting parameters with starting values.
Here we select starting values being quite far from true values
to puzzle our minimizer's as much as possible.
"""
params = ba.Parameters()
params.add("height", 1.*nm, min=0.01, max=30, step=0.05*nm)
params.add("radius", 20.*nm, min=0.01, max=30, step=0.05*nm)
"""
Now we run first minimization round using the Genetic minimizer.
The Genetic minimizer is able to explore large parameter space
without being trapped by some local minimum.
"""
minimizer = ba.Minimizer()
minimizer.setMinimizer("Genetic", "", "MaxIterations=2;RandomSeed=1")
result = minimizer.minimize(fit_objective.evaluate, params)
fit_objective.finalize(result)
best_params_so_far = result.parameters()
"""
Second fit will use another minimizer.
It starts from best parameters found in previous minimization
and then continues until fit converges.
"""
minimizer.setMinimizer("Minuit2", "Migrad")
result = minimizer.minimize(fit_objective.evaluate, best_params_so_far)
fit_objective.finalize(result)
if __name__ == '__main__':
run_fitting()
plt.show()
|
