fixed files form Math #13

Spirals-Team · Mar 7, 2017 · 5fef7b8 · 5fef7b8
1 parent 1211398
commit 5fef7b8
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/.../Math/13/org/apache/commons/math3/optimization/general/AbstractLeastSquaresOptimizer.java b/.../Math/13/org/apache/commons/math3/optimization/general/AbstractLeastSquaresOptimizer.java
@@ -559,7 +559,16 @@ protected double[] computeResiduals(double[] objectiveValue) {
      * @return the square-root of the weight matrix.
      */
     private RealMatrix squareRoot(RealMatrix m) {
+        if (m instanceof DiagonalMatrix) {
+            final int dim = m.getRowDimension();
+            final RealMatrix sqrtM = new DiagonalMatrix(dim);
+            for (int i = 0; i < dim; i++) {
+               sqrtM.setEntry(i, i, FastMath.sqrt(m.getEntry(i, i)));
+            }
+            return sqrtM;
+        } else {
             final EigenDecomposition dec = new EigenDecomposition(m);
             return dec.getSquareRoot();
+        }
     }
 }