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

keep all cat2 icings, keep some more no-icing

parent d8bb78f0
No related branches found
No related tags found
No related merge requests found
......@@ -526,13 +526,13 @@ def process(ice_dct, no_ice_dct, neg_ice_dct):
ice_keys_2 = np.array(ice_keys_2)
ice_tidx_2 = np.array(ice_tidx_2)
print('2: ', ice_keys_2.shape[0])
np.random.seed(42)
ridxs = np.random.permutation(np.arange(ice_keys_2.shape[0]))
ice_keys_2 = ice_keys_2[ridxs]
ice_tidx_2 = ice_tidx_2[ridxs]
ice_keys_2 = ice_keys_2[::2]
ice_tidx_2 = ice_tidx_2[::2]
print('2: reduced: ', ice_tidx_2.shape)
# np.random.seed(42)
# ridxs = np.random.permutation(np.arange(ice_keys_2.shape[0]))
# ice_keys_2 = ice_keys_2[ridxs]
# ice_tidx_2 = ice_tidx_2[ridxs]
# ice_keys_2 = ice_keys_2[::2]
# ice_tidx_2 = ice_tidx_2[::2]
# print('2: reduced: ', ice_tidx_2.shape)
ice_keys_1 = np.array(ice_keys_1)
ice_tidx_1 = np.array(ice_tidx_1)
......@@ -565,8 +565,8 @@ def process(ice_dct, no_ice_dct, neg_ice_dct):
ridxs = np.random.permutation(np.arange(no_ice_keys.shape[0]))
no_ice_keys = no_ice_keys[ridxs]
no_ice_tidx = no_ice_tidx[ridxs]
no_ice_keys = no_ice_keys[::20]
no_ice_tidx = no_ice_tidx[::20]
no_ice_keys = no_ice_keys[::16]
no_ice_tidx = no_ice_tidx[::16]
print('no ice reduced: ', no_ice_keys.shape[0])
sidxs = np.argsort(no_ice_keys)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment