code samples for dataset

emptymalei · Nov 27, 2023 · 4d6c710 · 4d6c710
1 parent b994e10
commit 4d6c710
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 5 deletions.
diff --git a/dl/notebooks/pendulum_dataset.py b/dl/notebooks/pendulum_dataset.py
@@ -17,14 +17,11 @@
 #
 # In this notebook, we demo a dataset we created to simulate the oscillations of a pendulumn.
 
-# +
 import matplotlib.pyplot as plt
 import pandas as pd
 from ts_dl_utils.datasets.dataset import DataFrameDataset
 from ts_dl_utils.datasets.pendulum import Pendulum, PendulumDataModule
 
-# -
-
 # ## Data
 #
 # We create a dataset that models a damped pendulum. The pendulum is modelled as a damped harmonic oscillator, i.e.,
@@ -60,8 +57,6 @@
 
 # ### PyTorch and Lighting DataModule
 
-# #### DataModule
-
 history_length = 100
 horizon = 5
 

diff --git a/dl/notebooks/ts_dl_utils/datasets/pendulum.py b/dl/notebooks/ts_dl_utils/datasets/pendulum.py
@@ -6,6 +6,7 @@
 import pandas as pd
 import torch
 from torch.utils.data import DataLoader, Dataset
+from ts_dl_utils.datasets.dataset import DataFrameDataset
 
 
 class Pendulum: