mirror of
https://github.com/webclinic017/drift.git
synced 2026-08-18 13:28:22 +00:00
feat(Forecasting): pytorch-forecasting scaffolding is now working, added narrow data format, fixed missing time column index name (#5)
* feat: Started implementing pytorch-forecasting. * feat(Forecasting): pytorch-forecasting scaffolding is now working, added narrow data format, fixed missing `time` column index name Co-authored-by: Daniel Szemerey <szemy2@gmail.com>
This commit is contained in:
co-authored by
Daniel Szemerey
parent
a59e705f02
commit
797791d2f4
@@ -0,0 +1,48 @@
|
||||
from load_data import load_files
|
||||
import pandas as pd
|
||||
# from tensorflow import keras
|
||||
from utils.normalize import normalize
|
||||
# import tensorflow as tf
|
||||
from utils.visualize import visualize_loss
|
||||
|
||||
from torch.utils.data import DataLoader, random_split
|
||||
from model_lightning import LitManualAutoEncoder
|
||||
import pytorch_lightning as pl
|
||||
|
||||
#%%
|
||||
data = load_files('data/', False)
|
||||
data.reset_index(drop=True, inplace=True)
|
||||
data = data[[column for column in data.columns if not column.endswith('volume')]]
|
||||
|
||||
data.head()
|
||||
|
||||
#%%
|
||||
ticker_to_predict = 'ETH_returns'
|
||||
|
||||
learning_rate = 0.002
|
||||
batch_size = 64
|
||||
epochs = 100
|
||||
|
||||
split_fraction = 0.715
|
||||
train_split = int(split_fraction * int(data.shape[0]))
|
||||
|
||||
past = 10
|
||||
future = 1
|
||||
|
||||
start = past + future
|
||||
end = start + train_split
|
||||
|
||||
|
||||
# train = DataLoader(train, batch_size=32)
|
||||
# test = DataLoader(test, batch_size=32)
|
||||
# val = DataLoader(val, batch_size=32)
|
||||
|
||||
|
||||
# init model
|
||||
ae = LitManualAutoEncoder()
|
||||
|
||||
# Initialize a trainer
|
||||
trainer = pl.Trainer(gpus=1, max_epochs=3, progress_bar_refresh_rate=20)
|
||||
|
||||
# Train the model ⚡
|
||||
# trainer.fit(ae, train, val)
|
||||
Reference in New Issue
Block a user