Caught errors from the mixed code of sparse/dense

Author: dongkwan-kim

research/gam/gam/data/dataset.py

@@ -545,7 +545,7 @@ def row_normalize(features):
     rowsum = np.array(features.sum(1))
     r_inv = np.power(rowsum, -1).flatten()
     r_inv[np.isinf(r_inv)] = 0.
-    r_mat_inv = scipy.diags(r_inv)
+    r_mat_inv = scipy.sparse.diags(r_inv)
     features = r_mat_inv.dot(features)
     return features
 

research/gam/gam/trainer/trainer_classification_gcn.py

@@ -21,8 +21,8 @@
 import logging
 import os
 
-from .adversarial_dense import entropy_y_x
-from .adversarial_dense import get_loss_vat
+from .adversarial_sparse import entropy_y_x
+from .adversarial_sparse import get_loss_vat
 import numpy as np
 import tensorflow as tf
 from .trainer_base import batch_iterator