USGS-R / river-dl

Deep learning model for predicting environmental variables on river systems
Creative Commons Zero v1.0 Universal
21 stars 15 forks source link

Make test/validation partitions optional #161

Closed jsadler2 closed 2 years ago

jsadler2 commented 2 years ago

Currently, the prep_all_data requires start and end dates for all three partitions (train, validation, test). I think it would make sense to have the validation and test sets be optional. If those are not defined, no data would be assigned to the x_val/y_val and/or x_tst/y_tst variables in the output data dictionary (the one that gets written to the .npz file).