Open nbren12 opened 5 years ago
@sarenehan
eta
uwnet.train.get_dataset
XRTimeSeries
setup_indices
time_length=2
ConditionalXrTimeSteps(data: xr.Dataset, conditional_state=2)
sample = dataset[0]
sample should be
sample
{ 'QT': (2, nz, 1, 1) 'stochastic_state': (2, 1, 1, 1) ... }
@sarenehan Since you are no longer training separate NNs for each bin, do you still need these data loader classes? If not, we could close this issue and delete the corresponding branch.
@sarenehan
eta
to the xarray training dataset inuwnet.train.get_dataset
XRTimeSeries
. changesetup_indices
and settime_length=2
.ConditionalXrTimeSteps(data: xr.Dataset, conditional_state=2)
.sample
should be