forked from simpeg/simpeg
/
Inversion.py
69 lines (50 loc) · 2.09 KB
/
Inversion.py
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
from __future__ import print_function
from SimPEG import Utils, sp, np
from .Optimization import Remember, IterationPrinters, StoppingCriteria
from . import Directives
class BaseInversion(object):
"""
Inversion Class.
"""
name = 'BaseInversion'
debug = False #: Print debugging information
counter = None #: Set this to a SimPEG.Utils.Counter() if you want to count things
@property
def directiveList(self):
if getattr(self,'_directiveList', None) is None:
self._directiveList = Directives.DirectiveList(inversion=self)
return self._directiveList
@directiveList.setter
def directiveList(self, value):
if type(value) is list:
value = Directives.DirectiveList(*value)
assert isinstance(value, Directives.DirectiveList), 'Must be a DirectiveList'
self._directiveList = value
self._directiveList.inversion = self
def __init__(self, invProb, directiveList=None, **kwargs):
if directiveList is None:
directiveList = []
self.directiveList = directiveList
Utils.setKwargs(self, **kwargs)
self.invProb = invProb
self.opt = invProb.opt
self.opt.callback = self._optCallback
self.stoppers = [StoppingCriteria.iteration]
# Check if we have inserted printers into the optimization
if IterationPrinters.phi_d not in self.opt.printers:
self.opt.printers.insert(1,IterationPrinters.beta)
self.opt.printers.insert(2,IterationPrinters.phi_d)
self.opt.printers.insert(3,IterationPrinters.phi_m)
@Utils.timeIt
def run(self, m0):
"""run(m0)
Runs the inversion!
"""
self.invProb.startup(m0)
self.directiveList.call('initialize')
print('curModel has any nan: {:b}'.format(np.any(np.isnan(self.invProb.curModel))))
self.m = self.opt.minimize(self.invProb.evalFunction, self.invProb.curModel)
self.directiveList.call('finish')
return self.m
def _optCallback(self, xt):
self.directiveList.call('endIter')