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

minor...

parent 027d11f1
Branches
No related tags found
No related merge requests found
......@@ -294,10 +294,14 @@ class IcingIntensityNN:
if trn_idxs is None:
time = self.h5f_trn['time']
trn_idxs = np.arange(time.shape[0])
if seed is not None:
np.random.seed(seed)
np.random.shuffle(trn_idxs)
time = self.h5f_tst['time']
tst_idxs = np.arange(time.shape[0])
if seed is not None:
np.random.seed(seed)
np.random.shuffle(tst_idxs)
self.num_data_samples = trn_idxs.shape[0]
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment