RelaxedNewton implemented, QuasiNewtonMinimizer has to be changed in _get_descent_direction

a6d099d6 · Jakob Knollmüller · 86e1eda4 · a6d099d6
Commit a6d099d6 authored Feb 06, 2017 by Jakob Knollmüller
Hide whitespace changes
Inline Side-by-side

Showing with 13 additions and 0 deletions

nifty/minimization/relaxed_newton.py nifty/minimization/relaxed_newton.py +13 -0

No files found.
--- a/nifty/minimization/relaxed_newton.py
+++ b/nifty/minimization/relaxed_newton.py
+# -*- coding: utf-8 -*-
+
+from .quasi_newton_minimizer import QuasiNewtonMinimizer
+
+
+class RelaxedNewton(QuasiNewtonMinimizer):
+    def _get_descend_direction(self, energy):
+        descend_direction = energy.curvature.inverse_times(energy.gradient)
+        norm = descend_direction.norm()
+        if norm != 1:
+            return descend_direction / -norm
+        else:
+            return descend_direction * -1