vincent-leguen / DILATE

Code for our NeurIPS 2019 paper "Shape and Time Distortion Loss for Training Deep Time Series Forecasting Models"

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Data Load

firmai opened this issue · comments

The dataloading stage is quite intense, is there possibly a way to showcase how a generaly pandas df with date index to be converted to fit your format here/

X_train_input,X_train_target,X_test_input,X_test_target,train_bkp,test_bkp = create_synthetic_dataset(N,N_input,N_output,sigma)
dataset_train = SyntheticDataset(X_train_input,X_train_target, train_bkp)
dataset_test  = SyntheticDataset(X_test_input,X_test_target, test_bkp)
trainloader = DataLoader(dataset_train, batch_size=batch_size,shuffle=True, num_workers=1)
testloader  = DataLoader(dataset_test, batch_size=batch_size,shuffle=False, num_workers=1)

I am also trying to use the datasets (ECG5000,...) from the paper to conduct the experiment. Were you able to do that @firmai ?

I am also trying to use the datasets (ECG5000,...) from the paper to conduct the experiment. Were you able to do that @firmai ?

I am also trying to use the datasets (ECG5000,...) from the paper to conduct the experiment. Were you able to do that @firmai ?