Update LFNA

2021-05-15 16:01:40 +08:00
parent b81ef2dd74
commit 72f240bf0a
12 changed files with 128 additions and 1050 deletions
--- a/lib/datasets/synthetic_env.py
+++ b/lib/datasets/synthetic_env.py
@@ -80,6 +80,12 @@ class SyntheticDEnv(data.Dataset):
    def timestamp_interval(self):
        return self._timestamp_generator.interval

+    def random_timestamp(self):
+        return (
+            random.random() * (self.max_timestamp - self.min_timestamp)
+            + self.min_timestamp
+        )
+
    def reset_max_seq_length(self, seq_length):
        self._seq_length = seq_length