USGS-R / river-dl

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

Specify train sites to remove from test metrics #201

Closed SimonTopp closed 2 years ago

SimonTopp commented 2 years ago

This PR adds the capability to remove specific training sites from your testing metrics in the case that train and test periods have overlapping years but distinct sites. Also changed on how the device in determined in predict_torch. In the previous implementation, it pulled the device based on the model parameters, but since model parameters can be spread across multiple devices it was ambiguous.