Changeset 82 for tests/test_mogi.py

Timestamp:

01/26/09 16:12:08 (7 years ago)

Author:

mmckerns

Message:

scipy.optimize.fmin fit to mystic.Solve interface;
mystic handlers, etc added to fmin solver;
VerboseSow? edited to handle simplex-type cloud of results per iteration

File:

• tests/test_mogi.py (modified) (3 diffs)

tests/test_mogi.py

@@ -110 +110 @@
 if __name__ == '__main__':
 
-    from mystic.nelder_mead import fmin
+    from mystic.scipy_optimize_fmin import NelderMeadSimplexSolver as fmin
+    from mystic.nmtools import IterationRelativeError as IRE
     from scipy.optimize import leastsq, fmin_cg
     #
@@ -125 +126 @@
     #
     simplex, esow = Sow(), Sow()
-    sol = fmin(cost_function, point,  EvaluationMonitor = esow, retall = True, StepMonitor = simplex)
-    print "simplex solution: ", sol[1][-1]
+    solver = fmin(len(point))
+    solver.SetInitialPoints(point)
+    killme = IRE(1e-4,1e-4)
+    solver.Solve(cost_function, killme, EvaluationMonitor = esow, StepMonitor = simplex)
+    sol = solver.Solution()
+
+    print "simplex solution: ", sol
     #
     solcg = fmin_cg(cost_function, point)
@@ -137 +143 @@
     plot_noisy_data()
     plot_sol(desol,'r-')
-    plot_sol(sol[1][-1],'k--')
+    plot_sol(sol,'k--')
     plot_sol(solcg,'b-.')
     pylab.legend(('Noisy data', 'Differential Evolution', 'Nelder Mead', 'Polak Ribiere'))