mirror of
https://github.com/webclinic017/drift.git
synced 2026-08-21 23:08:09 +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,18 @@
|
||||
import os
|
||||
import pandas as pd
|
||||
|
||||
import torch
|
||||
from torch.utils.data import Dataset
|
||||
|
||||
|
||||
class TimeSeriesDataset(Dataset):
|
||||
def __init__(self, file_loader_hook):
|
||||
|
||||
df = file_loader_hook()
|
||||
|
||||
def __len__(self):
|
||||
return len(self.img_labels)
|
||||
|
||||
def __getitem__(self, idx):
|
||||
pass
|
||||
# return image, label
|
||||
Reference in New Issue
Block a user