wassname/seq2seq-time
1
0
Fork 0
mirror of https://github.com/wassname/seq2seq-time.git synced 2026-09-12 12:51:10 +08:00
Code Issues Packages Projects Releases Wiki Activity

misc, fix transformer, more plots

Browse Source
This commit is contained in:
wassname
2020-10-30 14:46:31 +08:00
parent b7a80f187c
commit e1c86f9654
7 changed files with 1114 additions and 11017 deletions
Download Patch File Download Diff File Expand all files Collapse all files
seq2seq_time/predict.py
+1 -1
View File
@@ -49,7 +49,7 @@ def predict(model, ds_test, batch_size, device='cpu', scaler=None):
"y_true": (["t_source", "t_ahead",], y_future),
},
coords={"t_source": t_source, "t_ahead": t_ahead, "t_behind": t_behind},
attrs={'freq': ds_test.freq, "model": str(model), "targets": ds_test.columns_target}
attrs={'freq': str(ds_test.freq), "model": str(type(model)), "targets": ds_test.columns_target}
)
xrs.append(xr_out)