Exemple #1
0
    def run(self, r, niters=10000):
        """Run the lda kernel for `niters`, in a single thread.

        Parameters
        ----------
        r : random state
        niters : int

        """
        validator.validate_type(r, rng, param_name='r')
        validator.validate_positive(niters, param_name='niters')

        for _ in xrange(niters):
            for name, config in self._kernel_config:
                if name == 'crf':
                    lda_crp_gibbs(self._latent, r)
                elif name == 'direct_base_dp_hp':
                    sample_gamma(self._latent, r, config['hp1'], config['hp2'])
                elif name == 'direct_second_dp_hp':
                    sample_alpha(self._latent, r, config['hp1'], config['hp2'])
                elif name == 'direct_vocab_hp':
                    raise NotImplementedError(
                        'direct_vocab_hp not yet implemented')
                else:
                    raise ValueError(
                        "Bad kernel specification {}".format(name))
Exemple #2
0
    def run(self, r, niters=10000):
        """Run the lda kernel for `niters`, in a single thread.

        Parameters
        ----------
        r : random state
        niters : int

        """
        validator.validate_type(r, rng, param_name='r')
        validator.validate_positive(niters, param_name='niters')

        for _ in xrange(niters):
            lda_crp_gibbs(self._latent, r)
Exemple #3
0
    def run(self, r, niters=10000):
        """Run the lda kernel for `niters`, in a single thread.

        Parameters
        ----------
        r : random state
        niters : int

        """
        validator.validate_type(r, rng, param_name='r')
        validator.validate_positive(niters, param_name='niters')

        for _ in xrange(niters):
            lda_crp_gibbs(self._latent, r)
Exemple #4
0
    def run(self, r, niters=10000):
        """Run the lda kernel for `niters`, in a single thread.

        Parameters
        ----------
        r : random state
        niters : int

        """
        validator.validate_type(r, rng, param_name='r')
        validator.validate_positive(niters, param_name='niters')

        for _ in xrange(niters):
            for name, config in self._kernel_config:
                if name == 'crf':
                    lda_crp_gibbs(self._latent, r)
                elif name == 'direct_base_dp_hp':
                    sample_gamma(self._latent, r, config['hp1'], config['hp2'])
                elif name == 'direct_second_dp_hp':
                    sample_alpha(self._latent, r, config['hp1'], config['hp2'])
                elif name == 'direct_vocab_hp':
                    raise NotImplementedError('direct_vocab_hp not yet implemented')
                else:
                    raise ValueError("Bad kernel specification {}".format(name))