Update training and prediction code, add pretrained models.

itsluketwist · Aug 15, 2024 · e0ece43 · e0ece43
1 parent 4f274bf
commit e0ece43
Show file tree

Hide file tree

Showing 7 changed files with 91 additions and 878 deletions.
diff --git a/output/pretrained/GRU_state_data.pth b/output/pretrained/GRU_state_data.pth
diff --git a/output/pretrained/LSTM_state_data.pth b/output/pretrained/LSTM_state_data.pth
diff --git a/output/pretrained/RNN_state_data.pth b/output/pretrained/RNN_state_data.pth
diff --git a/output/pretrained/TCN_state_data.pth b/output/pretrained/TCN_state_data.pth
diff --git a/output/pretrained/TE_state_data.pth b/output/pretrained/TE_state_data.pth
diff --git a/predict.ipynb b/predict.ipynb
diff --git a/train.ipynb b/train.ipynb