Add paratition generator to utils

saghiles · Aug 10, 2019 · b69c6ba · b69c6ba
1 parent 56f1f3f
commit b69c6ba
Showing 1 changed file with 15 additions and 1 deletion.
diff --git a/utils.R b/utils.R
@@ -21,4 +21,18 @@ NMI <- function(rowcluster, truelabels, n){
     }
     resnmi = Num / (sqrt(denum_k * denum_l))
     resnmi
-}
+}
+
+# Partition generator
+par_gen<-
+  function(n,k, nb_par = 1){
+
+            par = as.integer(sample(as.numeric(1:k),n,replace = TRUE))
+            if(nb_par >1){
+              for(i in 2:nb_par){
+                par = rbind(par,as.integer(sample(as.numeric(1:k),n,replace = TRUE)))
+              }
+              par = as.matrix(par)
+            }
+            par
+  }