def setup()

in flows.py [0:0]

• 28 lines of code
• 4 McCabe index (conditional complexity)

    def setup(self):
        if self.cost_gamma == 'None': self.cost_gamma = None
        if self.min_zero_gamma == 'None': self.min_zero_gamma = None

        mu_init = densities.get_uniform(self.manifold)

        self.mus = []
        self.alphas = []
        self.ws = []
        input_sz = self.manifold.D
        for i in range(self.n_layers):

            key = f'mu{i:02d}'
            mu = self.param(
                key, init_manifold_samples(mu_init),
                [self.manifold.D, self.n_components])
            setattr(self, key, mu)

            key = f'alpha{i:02d}'
            alpha = self.param(
            key, init_uniform(
                minval=self.init_alpha_minval,
                maxval=self.init_alpha_minval+self.init_alpha_range),
            [self.n_components])
            setattr(self, key, alpha)

            key = f'w{i:02d}'
            w = self.param(
                key, init_uniform(minval=0., maxval=1.), [1])
            setattr(self, key, w)

            self.mus.append(mu)
            self.alphas.append(alpha)
            self.ws.append(w)