def __init__()

in mico/utils/net_utils.py [0:0]

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

    def __init__(self, number_clusters):
        """This is the function approximating the distribution of cluster sizes for document assignment.
        We use `softmax` to make sure the output is a distribution.

        Parameters
        ----------
        number_clusters : int
            The number of clusters to which we are going to assign the documents.

        """
        super(MarginalDistributionZ, self).__init__()
        self.softmax = nn.Softmax(dim=-1)
        self.theta = nn.Embedding(1, number_clusters)