class_docstr: Tests initializations for NUTS sampler on models
class_docstr_tok: ["Tests","initializations","for","NUTS","sampler","on","models"]
code: def track_marginal_mixture_model_ess(self, init):
      model, start = mixture_model()
      with model:
          _, step = pm.init_nuts(
              init=init,
              chains=self.chains,
              progressbar=False,
              random_seed=np.arange(self.chains),
          )
          start = [
              {k: v for k, v in start.items()}
              for _ in range(self.chains)
          ]
          t0 = time.time()
          idata = pm.sample(
              draws=self.draws,
              step=step,
              cores=4,
              chains=self.chains,
              start=start,
              seeds=np.arange(self.chains),
              progressbar=False,
              compute_convergence_checks=False,
          )
          tot = time.time() - t0
      ess = az.ess(idata, var_names=["mu"])[
          "mu"
      ].values.min()  # worst case
      return ess / tot
  
imports: import time
  import timeit
  import arviz
  import numpy
  import pandas
  import pytensor
  import pytensor.tensor
  import pymc
label_desc: Use image node_id 3 for calling the NUTSInitSuite obj's underlying member method code with example usage: obj.track_marginal_mixture_model_ess(init) and returns: unknown