Method fit

crates/openquant/src/ef3m.rs:209–249

(&mut self, mut mu_2: f64)

    }

    pub fn fit(&mut self, mut mu_2: f64) -> Result<(), String> {
        let mut rng = rand::thread_rng();
        let mut p_1 = rng.gen_range(0.0..1.0);
        let mut num_iter = 0usize;

        loop {
            num_iter += 1;
            let parameters_new = match self.variant {
                1 => self.iter_4(mu_2, p_1),
                2 => self.iter_5(mu_2, p_1),
                _ => return Err("Value of argument 'variant' must be either 1 or 2.".to_string()),
            };
            if parameters_new.is_empty() {
                return Ok(());
            }

            let parameters = parameters_new.clone();
            let _ = self.get_moments(&parameters, false);
            let error: f64 = self
                .moments
                .iter()
                .zip(self.new_moments.iter())
                .map(|(a, b)| (a - b).powi(2))
                .sum();
            if error < self.error {
                self.parameters = parameters.clone();
                self.error = error;
            }

            if (p_1 - parameters[4]).abs() < self.epsilon {
                self.parameters = parameters;
                break;
            }
            if num_iter > self.max_iter {
                return Ok(());
            }
            p_1 = parameters[4];
            mu_2 = parameters[1];
        }
        Ok(())
    }

    pub fn single_fit_loop(&mut self, epsilon_override: Option<f64>) -> Vec<FitResultRow> {
        if let Some(eps) = epsilon_override {