Sadjad Alikhani
commited on
Commit
•
573514a
1
Parent(s):
78d5782
Update input_preprocess.py
Browse files- input_preprocess.py +2 -2
input_preprocess.py
CHANGED
@@ -246,8 +246,8 @@ def make_sample(user_idx, patch, word2id, n_patches, n_masks, patch_size, gen_ra
|
|
246 |
input_ids[pos] = np.random.rand(patch_size)
|
247 |
elif rnd_num < 0.9:
|
248 |
input_ids[pos] = word2id['[MASK]']
|
249 |
-
if user_idx == 0:
|
250 |
-
|
251 |
return [input_ids, masked_tokens, masked_pos]
|
252 |
|
253 |
|
|
|
246 |
input_ids[pos] = np.random.rand(patch_size)
|
247 |
elif rnd_num < 0.9:
|
248 |
input_ids[pos] = word2id['[MASK]']
|
249 |
+
#if user_idx == 0:
|
250 |
+
# print(f'masked_pos: {masked_pos}')
|
251 |
return [input_ids, masked_tokens, masked_pos]
|
252 |
|
253 |
|