Esempio n. 1
0
        # kmc sampler instance
        kmc = KMCStatic(surrogate, target, momentum, num_steps, num_steps,
                        step_size, step_size)

        # simulate trajectory from starting point, note _proposal_trajectory is a "hidden" method
        Qs_total = []
        acc_probs_total = []
        accor_total = []
        ksd_total = []
        ess_total = []
        mean_x1_total = []
        np.random.seed(seed + 1)
        for i in xrange(M):
            current = start_samples[i]
            current_log_pdf = target.log_pdf(current)
            Qs, acc_probs, log_pdf_q = kmc._proposal_trajectory(
                current, current_log_pdf)
            # compute auto correlation on first dim
            accor = autocorr(Qs[:, 0])
            accor_total.append(accor)
            Qs_total.append(Qs)
            # compute min ESS
            T_ess = 1800
            ess = RCodaTools.ess_coda_vec(Qs[T_ess + 1:])
            ess = np.minimum(ess, Qs[T_ess + 1:].shape[0])
            min_ess = np.min(ess)
            ess_total.append(min_ess)
            # compute acceptance prob
            acc_probs_total.append(acc_probs)
            # compute E[x1] estimates for different time t