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

keep more data in process_2

parent baa12f15
Branches
No related tags found
No related merge requests found
......@@ -419,7 +419,7 @@ def process_2(ice_dct, no_ice_dct, neg_ice_dct):
print('2: ', ice_keys_2.shape)
np.random.seed(42)
np.random.shuffle(ice_keys_2)
ice_keys_2 = ice_keys_2[0:30000]
ice_keys_2 = ice_keys_2[0:60000]
ice_keys_1 = np.array(ice_keys_1)
print('1: ', ice_keys_1.shape)
......@@ -437,7 +437,7 @@ def process_2(ice_dct, no_ice_dct, neg_ice_dct):
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]
no_ice_keys = no_ice_keys[0:150000]
uniq_sorted_no_ice = np.unique(no_ice_keys)
print('no ice: ', no_ice_keys.shape, uniq_sorted_no_ice.shape)
......@@ -449,7 +449,7 @@ def process_2(ice_dct, no_ice_dct, neg_ice_dct):
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]
neg_ice_keys = neg_ice_keys[0:10000]
uniq_sorted_neg_ice = np.unique(neg_ice_keys)
print('neg ice: ', neg_ice_keys.shape, uniq_sorted_neg_ice.shape)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment