Commit 786f0c8c authored by lukas leufen's avatar lukas leufen 👻
Browse files

update dummy data handler

parent 59f202ba
Pipeline #101705 passed with stages
in 10 minutes and 58 seconds
......@@ -106,6 +106,9 @@ class DummyData:
Y2 = np.random.randint(21, 30, size=(self.number_of_samples, 5, 1)) # samples, window, variables
return [Y1, Y2]
def get_data(self, upsampling=False, as_numpy=True):
return self.get_X(upsampling, as_numpy), self.get_Y(upsampling, as_numpy)
class TestKerasIterator:
......
......@@ -150,3 +150,6 @@ class DummyData:
Y1 = np.random.randint(0, 10, size=(self.number_of_samples, 5)) # samples, window
Y2 = np.random.randint(21, 30, size=(self.number_of_samples, 3)) # samples, window
return [Y1, Y2]
def get_data(self, upsampling=False, as_numpy=True):
return self.get_X(upsampling, as_numpy), self.get_Y(upsampling, as_numpy)
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment