mirror of
https://github.com/webclinic017/drift.git
synced 2026-08-04 06:37:45 +00:00
Mark Aron Szulyovszky
797791d2f4
* 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>
18 lines
335 B
Python
18 lines
335 B
Python
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 |