Merge pull request #536 from bbayukari/master

Mamba413 · web-flow · commit ae84e8927976 · 2024-05-30T11:45:00.000+08:00
[BUG] When support set is range(k)
diff --git a/src/Algorithm.h b/src/Algorithm.h
@@ -438,7 +438,7 @@ class Algorithm {
             }
 
             // If A_U not change, U will not change and we can stop.
-            if (A_U.size() == 0 || A_U.maxCoeff() == T0 - 1) break;
+            if (this->U_size < N && (A_U.size() == 0 || A_U.maxCoeff() == T0 - 1)) break;
 
             // Update & Restore beta, A from U
             slice_restore(beta_U, U_ind, beta);

Original file line number	Diff line number	Diff line change
`@@ -438,7 +438,7 @@ class Algorithm {`
`438`	`438`	`}`
`439`	`439`
`440`	`440`	`// If A_U not change, U will not change and we can stop.`
`441`		`- if (A_U.size() == 0 \|\| A_U.maxCoeff() == T0 - 1) break;`
	`441`	`+ if (this->U_size < N && (A_U.size() == 0 \|\| A_U.maxCoeff() == T0 - 1)) break;`
`442`	`442`
`443`	`443`	`// Update & Restore beta, A from U`
`444`	`444`	`slice_restore(beta_U, U_ind, beta);`