You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
66 lines
2.0 KiB
66 lines
2.0 KiB
import logging as log |
|
|
|
import pandas as pd |
|
from typing import Optional |
|
|
|
from detectors import Detector |
|
from models import ModelCache |
|
from time import time |
|
|
|
|
|
logger = log.getLogger('THRESHOLD_DETECTOR') |
|
|
|
|
|
class ThresholdDetector(Detector): |
|
|
|
def __init__(self): |
|
pass |
|
|
|
def train(self, dataframe: pd.DataFrame, threshold: dict, cache: Optional[ModelCache]) -> ModelCache: |
|
log.debug('run train for threshold detector') |
|
return { |
|
'cache': { |
|
'value': threshold['value'], |
|
'condition': threshold['condition'] |
|
} |
|
} |
|
|
|
def detect(self, dataframe: pd.DataFrame, cache: Optional[ModelCache]) -> dict: |
|
log.debug('run detect for threshold detector') |
|
value = cache['value'] |
|
condition = cache['condition'] |
|
|
|
dataframe_without_nans = dataframe.dropna() |
|
if len(dataframe_without_nans) == 0: |
|
return dict() |
|
last_entry = dataframe_without_nans.iloc[-1] |
|
last_value = last_entry['value'] |
|
|
|
now = int(time()) * 1000 |
|
segment = ({ 'from': now, 'to': now }) |
|
segments = [] |
|
if condition == '>': |
|
if last_value > value: |
|
segments.append(segment) |
|
elif condition == '>=': |
|
if last_value >= value: |
|
segments.append(segment) |
|
elif condition == '=': |
|
if last_value == value: |
|
segments.append(segment) |
|
elif condition == '<=': |
|
if last_value <= value: |
|
segments.append(segment) |
|
elif condition == '<': |
|
if last_value < value: |
|
segments.append(segment) |
|
log.debug('seg {}'.format(segments)) |
|
return { |
|
'cache': cache, |
|
'segments': segments, |
|
'lastDetectionTime': now |
|
} |
|
|
|
def recieve_data(self, data: pd.DataFrame, cache: Optional[ModelCache]) -> Optional[dict]: |
|
log.debug('threshhold recieve data') |
|
return self.detect(data, cache)
|
|
|