fix:More robust normalization.

CNS1mple · Nov 22, 2021 · e4e1d05 · e4e1d05
1 parent b2c6475
commit e4e1d05
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/NonNegativeMatrixFactorization.m b/NonNegativeMatrixFactorization.m
@@ -332,7 +332,7 @@ Mathematica is (C) Copyright 1988-2013 Wolfram Research, Inc.
     Block[{d, S, SI},
       d = Table[Norm[H[[i]]], {i, Length[H]}];
       S = DiagonalMatrix[d];
-      SI = DiagonalMatrix[1 / d];
+      SI = DiagonalMatrix[Map[If[# != 0, 1 / #, 0]&, d]];
       {W.S, SI.H}
     ];