wassname/simpeg
1
0
Fork 0
mirror of https://github.com/wassname/simpeg.git synced 2026-07-07 12:34:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

Updates to adjoint and derivative tests

Browse Source
This commit is contained in:
rowanc1
2014-07-03 13:39:51 -07:00
parent 58a1101448
commit 10a39bc10c
2 changed files with 30 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files
simpegDC/BaseDC.py
+16 -9
View File
@@ -161,6 +161,10 @@ class ProblemDC(Problem.BaseProblem):
if u is None:
# Run forward simulation if $u$ not provided
u = self.fields(self.curModel)
else:
shp = (self.mesh.nC, self.survey.nTx)
u = u.reshape(shp, order='F')
D = self.mesh.faceDiv
G = self.mesh.cellGrad
# Derivative of inner product, $\left(\mathbf{M}_{1/\sigma}^f\right)^{-1}$
@@ -186,26 +190,29 @@ class ProblemDC(Problem.BaseProblem):
def Jtvec(self, m, v, u=None):
"""Takes data, turns it into a model..ish"""
self.curModel = m
sigma = self.curModel.transform # $\sigma = \mathcal{M}(\m)$
if u is None:
u = self.fields(m)
u = self.survey.reshapeFields(u)
v = self.survey.reshapeFields(v)
u = self.fields(self.curModel)
shp = (self.mesh.nC, self.survey.nTx)
u = u.reshape(shp, order='F')
P = self.survey.getP(self.mesh)
PT_x_v = (P.T*v).reshape(shp, order='F')
D = self.mesh.faceDiv
G = self.mesh.cellGrad
A = self.getA(m)
Av_dm = self.mesh.getFaceInnerProductDeriv(m)
A = self.A
Av_dm = self.mesh.getFaceInnerProductDeriv(sigma, invProp=True, invMat=True)
mT_dm = self.mapping.deriv(m)
dCdu = A.T
Ainv = self.Solver(dCdu)
w = Ainv * (P.T*v)
w = Ainv * PT_x_v
Jtv = 0
for i, ui in enumerate(u.T): # loop over each column
Jtv += Utils.sdiag( G * ui ) * ( D.T * w[:,i] )
Jtv += Av_dm( G * ui ).T * ( D.T * w[:,i] )
Jtv = - mT_dm.T * ( Av_dm.T * Jtv )
Jtv = - mT_dm.T * ( Jtv )
return Jtv
simpegDC/Tests/test_forward_DCproblem.py
+14 -14
View File
@@ -33,21 +33,21 @@ class DCProblemTests(unittest.TestCase):
passed = Tests.checkDerivative(derChk, self.m0, plotIt=False)
self.assertTrue(passed)
# def test_adjoint(self):
# # Adjoint Test
# u = np.random.rand(self.mesh.nC*self.survey.RHS.shape[1])
# v = np.random.rand(self.mesh.nC)
# w = np.random.rand(self.survey.dobs.shape[0])
# wtJv = w.dot(self.p.Jvec(self.m0, v, u=u))
# vtJtw = v.dot(self.p.Jtvec(self.m0, w, u=u))
# passed = np.abs(wtJv - vtJtw) < 1e-10
# print 'Adjoint Test', np.abs(wtJv - vtJtw), passed
# self.assertTrue(passed)
def test_adjoint(self):
# Adjoint Test
u = np.random.rand(self.mesh.nC*self.survey.nTx)
v = np.random.rand(self.mesh.nC)
w = np.random.rand(self.survey.dobs.shape[0])
wtJv = w.dot(self.p.Jvec(self.m0, v, u=u))
vtJtw = v.dot(self.p.Jtvec(self.m0, w, u=u))
passed = np.abs(wtJv - vtJtw) < 1e-10
print 'Adjoint Test', np.abs(wtJv - vtJtw), passed
self.assertTrue(passed)
# def test_dataObj(self):
# derChk = lambda m: [self.dmis.eval(m), self.dmis.evalDeriv(m)]
# passed = Tests.checkDerivative(derChk, self.m0, plotIt=False)
# self.assertTrue(passed)
def test_dataObj(self):
derChk = lambda m: [self.dmis.eval(m), self.dmis.evalDeriv(m)]
passed = Tests.checkDerivative(derChk, self.m0, plotIt=False)
self.assertTrue(passed)
if __name__ == '__main__':