change convergence criteria

R/fit_h.R

@@ -12,9 +12,13 @@
 #' @param min_mutations_number num: (type double) minimum number of accepted mutations for a segment to be included in the inference
 #' @param n_components  number of components specified from user 
 #'
+#' @param initial_iter description
+#' @param grad_samples description
+#' @param elbo_samples description
+#'
 #' @return   results_and_data = list(data = input_data_list, results = results, output_files_list = output_files_list)
 #' @export
-fit_h = function(x, max_attempts=2, INIT=TRUE, tolerance = 0.01, possible_k = c("2:1", "2:2", "2:0"), alpha = .05, min_mutations_number = 2, n_components = 0)
+fit_h = function(x, max_attempts=2, INIT=TRUE, tolerance = 0.01, possible_k = c("2:1", "2:2", "2:0"), alpha = .05, min_mutations_number = 2, n_components = 0, initial_iter=200, grad_samples=200, elbo_samples=200)
 {
   # stopifnot(inherits(x, 'cnaqc'))
 
@@ -73,9 +77,9 @@ fit_h = function(x, max_attempts=2, INIT=TRUE, tolerance = 0.01, possible_k = c(
                                initialization = inits_chain,
                                max_attempts = max_attempts,
                                INIT = INIT,
-                               initial_iter = 1000,
-                               grad_samples = 10,
-                               elbo_samples = 100,
+                               initial_iter = initial_iter,
+                               grad_samples = grad_samples,
+                               elbo_samples = elbo_samples,
                                tolerance = tolerance)
     }