Upgrade LFNA

2021-05-13 13:42:33 +00:00
parent cfabd05de8
commit c2fa181bc5
3 changed files with 5 additions and 2 deletions
--- a/lib/datasets/synthetic_env.py
+++ b/lib/datasets/synthetic_env.py
@@ -110,8 +110,10 @@ class SyntheticDEnv(data.Dataset):
        if self._seq_length is None:
            return self.__call__(timestamp)
        else:
+            noise = random.random() * self.timestamp_interval * 0.3
            timestamps = [
-                timestamp + i * self.timestamp_interval for i in range(self._seq_length)
+                timestamp + i * self.timestamp_interval + noise
+                for i in range(self._seq_length)
            ]
            xdata = [self.__call__(timestamp) for timestamp in timestamps]
            return zip_sequence(xdata)