wassname/attentive-neural-processes
1
0
Fork 0
mirror of https://github.com/wassname/attentive-neural-processes.git synced 2026-06-27 18:03:39 +08:00
Code Issues Packages Projects Releases Wiki Activity

MCDropout

Browse Source
This commit is contained in:
wassname
2020-02-16 21:40:39 +08:00
parent 315edbee8e
commit 2ec7b99bb7
3 changed files with 3386 additions and 118 deletions
Download Patch File Download Diff File Expand all files Collapse all files
smartmeters-ANP-RNN-mcdropout.ipynb
+253 -117
View File
File diff suppressed because one or more lines are too long
smartmeters-ANP-RNN.ipynb
+3132
View File
File diff suppressed because it is too large Load Diff
src/models/lstm_std.py
+1 -1
View File
@@ -272,7 +272,7 @@ class LSTM_PL(pl.LightningModule):
return parser
def plot_from_loader(loader, model, vis_i=670, n=1):
def plot_from_loader(loader, model, vis_i=670, n=1, window_len=0):
dset_test = loader.dataset
label_names = dset_test.label_names
y_trues = []