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.
29 lines
864 B
29 lines
864 B
import config |
|
import detectors |
|
import logging |
|
|
|
|
|
logger = logging.getLogger('AnalyticUnitWorker') |
|
|
|
|
|
class AnalyticUnitWorker: |
|
|
|
def __init__(self, analytic_unit_id: str, detector: detectors.Detector): |
|
self.analytic_unit_id = analytic_unit_id |
|
self.detector = detector |
|
|
|
async def do_learn(self, analytic_unit_id, payload) -> None: |
|
pattern = payload['pattern'] |
|
segments = payload['segments'] |
|
data = payload['data'] # [time, value][] |
|
await self.detector.train(data, segments) |
|
|
|
async def do_predict(self, analytic_unit_id, payload): |
|
pattern = payload['pattern'] |
|
data = payload['data'] # [time, value][] |
|
|
|
segments, last_prediction_time = await self.detector.predict(data) |
|
return { |
|
'segments': segments, |
|
'lastPredictionTime': last_prediction_time |
|
}
|
|
|