コード例 #1
0
 def _symmetrize_matrix_and_fix_diagonal(
         x_non_symmetric: csr_matrix) -> csr_matrix:
     x_symmetric = x_non_symmetric.tolil()
     r, c = x_symmetric.nonzero()
     x_symmetric[c, r] = x_symmetric[r, c]
     r = np.arange(x_symmetric.shape[0])
     x_symmetric[r, r] = 1
     return x_symmetric.tocsr()