Improved automatic subsampling size calculation.

1 files changed, 1 insertions, 1 deletions

diff --git a/main.cpp b/main.cpp
index 4aad1db..a0d20c8 100644
--- a/main.cpp
+++ b/main.cpp
@@ -123,7 +123,7 @@ int main(int argc, char **argv)
     arma::vec labels = m->labels();
 
     arma::uword n = X.n_rows;
-    int subsampleSize = 3 * ((int) sqrt(n));
+    int subsampleSize = (int) (3 * sqrt(n));
     arma::uvec sampleIndices;
     arma::mat Ys;