Skip to content
Snippets Groups Projects
Commit 92ea5dfc authored by tomrink's avatar tomrink
Browse files

snapshot...

parent dbe1cf8a
Branches
No related tags found
No related merge requests found
......@@ -414,6 +414,7 @@ def process_2(ice_dct, no_ice_dct, neg_ice_dct):
new_ice_dct[key] = ice_dct[key]
no_ice_keys = np.array(no_ice_keys)
print('no ice total: ', no_ice_keys.shape)
np.random.seed(42)
np.random.shuffle(no_ice_keys)
no_ice_keys = no_ice_keys[0:50000]
......@@ -425,6 +426,7 @@ def process_2(ice_dct, no_ice_dct, neg_ice_dct):
new_no_ice_dct[key] = no_ice_dct[key]
neg_ice_keys = np.array(neg_ice_keys)
print('neg ice total: ', neg_ice_keys.shape)
np.random.seed(42)
np.random.shuffle(neg_ice_keys)
neg_ice_keys = neg_ice_keys[0:5000]
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment