Skip to content
Snippets Groups Projects
Commit fa0f7a90 authored by Frisinghelli Daniel's avatar Frisinghelli Daniel
Browse files

Change train-test ratio to 1/10.

parent 7566300c
No related branches found
No related tags found
No related merge requests found
...@@ -121,10 +121,11 @@ if __name__ == '__main__': ...@@ -121,10 +121,11 @@ if __name__ == '__main__':
wet_days = (Obs_ds.sel(time=CALIB_PERIOD).mean(dim=('y', 'x')) >= wet_days = (Obs_ds.sel(time=CALIB_PERIOD).mean(dim=('y', 'x')) >=
WET_DAY_THRESHOLD).to_array().values.squeeze() WET_DAY_THRESHOLD).to_array().values.squeeze()
train, valid = train_test_split( train, valid = train_test_split(
CALIB_PERIOD, stratify=wet_days, test_size=0.5) CALIB_PERIOD, stratify=wet_days, test_size=0.1)
train, valid = sorted(train), sorted(valid) # sort chronologically train, valid = sorted(train), sorted(valid) # sort chronologically
else: else:
train, valid = train_test_split(CALIB_PERIOD, shuffle=False) train, valid = train_test_split(CALIB_PERIOD, shuffle=False,
test_size=0.1)
# training and validation dataset # training and validation dataset
Era5_train, Obs_train = Era5_ds.sel(time=train), Obs_ds.sel(time=train) Era5_train, Obs_train = Era5_ds.sel(time=train), Obs_ds.sel(time=train)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment