1 files changed, 30 insertions, 0 deletions
diff --git a/gi/posterior-regularisation/prjava/src/optimization/stopCriteria/GradientL2Norm.java b/gi/posterior-regularisation/prjava/src/optimization/stopCriteria/GradientL2Norm.java
new file mode 100644
index 00000000..534ff833
--- /dev/null
+++ b/gi/posterior-regularisation/prjava/src/optimization/stopCriteria/GradientL2Norm.java
@@ -0,0 +1,30 @@
+package optimization.stopCriteria;
+
+import optimization.gradientBasedMethods.Objective;
+import optimization.util.MathUtils;
+
+public class GradientL2Norm implements StopingCriteria{
+	
+	/**
+	 * Stop if gradientNorm/(originalGradientNorm) smaller
+	 * than gradientConvergenceValue
+	 */
+	protected double gradientConvergenceValue;
+	
+	
+	public GradientL2Norm(double gradientConvergenceValue){
+		this.gradientConvergenceValue = gradientConvergenceValue;
+	}
+	
+	public void reset(){}
+	
+	public boolean stopOptimization(Objective obj){
+		double norm = MathUtils.L2Norm(obj.gradient);
+		if(norm < gradientConvergenceValue){
+			System.out.println("Gradient norm below treshold");
+			return true;
+		}
+		return false;
+		
+	}
+}