wrapped prenet in a function in prenet_predict.py

sqrhussain · Dec 19, 2023 · 80999ab · 80999ab
1 parent 754bbb3
commit 80999ab
Show file tree

Hide file tree

Showing 6 changed files with 553 additions and 466 deletions.
diff --git a/data_interpolation.py b/data_interpolation.py
@@ -10,8 +10,8 @@
 from scipy.sparse import vstack, csc_matrix
 from sklearn.neighbors import KDTree
 
-def inject_noise_sparse(seed, n_out):     
-    rng = np.random.RandomState(42) 
+def inject_noise_sparse(seed, n_out):
+    rng = np.random.RandomState(42)
     n_sample, dim = seed.shape
     swap_ratio = 0.05
     n_swap_feat = int(swap_ratio * dim)
@@ -27,8 +27,8 @@ def inject_noise_sparse(seed, n_out):
         noise[i, swap_feats] = o2[0, swap_feats]
     return noise.tocsr()
 
-def inject_noise(seed, n_out):     
-    rng = np.random.RandomState(42) 
+def inject_noise(seed, n_out):
+    rng = np.random.RandomState(42)
     n_sample, dim = seed.shape
     swap_ratio = 0.05
     n_swap_feat = int(swap_ratio * dim)