diff --git a/average.py b/average.py index a7bb3fc..67d1eea 100644 --- a/average.py +++ b/average.py @@ -8,7 +8,9 @@ class step(algorithm): self.config_dict_ = None self.config_ = None self.average_number_ = 10 - self.window_1_ = None + self.present_number_ = 0 + self.window_1_ = [] + self.window_2_ = [] def set_config(self, config): self.config_ = config @@ -19,4 +21,10 @@ class step(algorithm): return self.config_ def eval(self, value): - return value * self.amplitude_base_ + self.present_number_ = len(self.window_1_) + if self.present_number_ < self.average_number_: + self.window_1_.append(value) + self.window_2_.append(value) + return + else: + return value * self.amplitude_base_ diff --git a/average_func_test.py b/average_func_test.py index cace41e..2cdb425 100644 --- a/average_func_test.py +++ b/average_func_test.py @@ -1,3 +1,10 @@ from algorithm import algorithm import json -import ray \ No newline at end of file +import ray +import numpy as np +a = np.zeros(shape=(5,2)) +b = [] +print(a) +print(type(b)) +print(len(b)) +