Skip to content
Snippets Groups Projects
Commit 9b257830 authored by tomrink's avatar tomrink
Browse files

minor...

parent 22ac47ae
Branches
No related tags found
No related merge requests found
......@@ -16,6 +16,36 @@ class GenericException(Exception):
self.message = message
class EarlyStop:
def __init__(self, window_length=3, patience=5):
self.patience = patience
self.min = np.finfo(np.single)
self.cnt = 0
self.cnt_wait = 0
self.window = np.zeros(window_length, dtype=np.single)
self.window.fill(np.nan)
def check_stop(self, value):
self.window[:-1] = self.window[1:]
self.window[-1] = value
if np.any(np.isnan(self.window)):
return False
ave = np.mean(self.window)
if ave < self.min:
self.min = ave
self.cnt_wait = 0
return False
else:
self.cnt_wait += 1
if self.cnt_wait > self.patience:
return True
else:
return False
def get_time_tuple_utc(timestamp):
dt_obj = datetime.datetime.fromtimestamp(timestamp, timezone.utc)
return dt_obj, dt_obj.timetuple()
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment