graphlet_kernel.py 文件源码

def generate_permutation_matrix():
    P = np.zeros((2**15,2**15),dtype=np.uint8)

    for a in range(2):
        for b in range(2):
            for c in range(2):
                for d in range(2):
                    for e in range(2):
                        for f in range(2):
                            for g in range(2):
                                for h in range(2):
                                    for i in range(2):
                                        for j in range(2):
                                            for k in range(2):
                                                for l in range(2):
                                                    for m in range(2):
                                                        for n in range(2):
                                                            for o in range(2):  
                                                                A = np.array([[0,a,b,c,d,e],[a,0,f,g,h,i],[b,f,0,j,k,l],[c,g,j,0,m,n],[d,h,k,m,0,o],[e,i,l,n,o,0]])

                                                                perms = multiset_permutations(np.array(range(6),dtype=np.uint8))
                                                                Per = np.zeros((factorial(6),6),dtype=np.uint8)
                                                                ind = 0
                                                                for permutation in perms:
                                                                    Per[ind,:] = permutation
                                                                    ind += 1

                                                                for p in range(factorial(6)):
                                                                    A_per = A[np.ix_(Per[p,:],Per[p,:])]
                                                                    P[graphlet_type(A), graphlet_type(A_per)] = 1
    return P