Update input_preprocess.py

input_preprocess.py

@@ -225,6 +225,7 @@ def make_sample(user_idx, patch, word2id, n_patches, n_masks, patch_size, gen_ra
     Returns:
         sample (list): Generated sample for the user.
     """
+    set_random_seed()
     
     tokens = patch[user_idx]
     input_ids = np.vstack((word2id['[CLS]'], tokens))
@@ -244,7 +245,7 @@ def make_sample(user_idx, patch, word2id, n_patches, n_masks, patch_size, gen_ra
                 input_ids[pos] = np.random.rand(patch_size)
             elif rnd_num < 0.9:
                 input_ids[pos] = word2id['[MASK]']
-
+    print(f'masked_pos: {masked_pos}')
     return [input_ids, masked_tokens, masked_pos]