affinity_mapping.py 文件源码

def compute(self, config):

        INPUT_ITR = self.iterator_batch(self._iterator_mean_cluster_vectors())
        Z = self.cluster_affinity_states(INPUT_ITR, size=self.cluster_n)

        print("Initial affinity grouping", Z.shape)
        # print self.vocab_n, self.cluster_n

        INPUT_ITR = self.iterator_batch(Z)
        Z2 = self.cluster_affinity_states(INPUT_ITR, size=len(Z))

        print("Final affinity size", len(Z2))
        self.save(config, Z2)

        '''
        import seaborn as sns
        plt = sns.plt
        DZ2 = cdist(Z2,Z2,metric='cosine')
        sns.heatmap(DZ2,xticklabels=False, yticklabels=False,linewidths=0)
        sns.plt.figure()
        #plt.show()

        DZ = cdist(Z,Z,metric='cosine')
        sns.heatmap(DZ,xticklabels=False, yticklabels=False,linewidths=0)
        #sns.plt.figure()
        sns.plt.show()
        '''

        self.h5.close()