Browse Source

fix (#737)

pull/1/head
Evgeny Smyshlyaev 5 years ago committed by Alexey Velikiy
parent
commit
0363558b91
  1. 3
      analytics/analytics/detectors/pattern_detector.py

3
analytics/analytics/detectors/pattern_detector.py

@ -42,7 +42,8 @@ class PatternDetector(Detector):
def __init__(self, pattern_type: str, analytic_unit_id: AnalyticUnitId): def __init__(self, pattern_type: str, analytic_unit_id: AnalyticUnitId):
super().__init__(analytic_unit_id) super().__init__(analytic_unit_id)
self.model = resolve_model_by_pattern(pattern_type) self.pattern_type = pattern_type
self.model = resolve_model_by_pattern(self.pattern_type)
self.bucket = DataBucket() self.bucket = DataBucket()
def train(self, dataframe: pd.DataFrame, segments: List[dict], cache: Optional[ModelCache]) -> ModelCache: def train(self, dataframe: pd.DataFrame, segments: List[dict], cache: Optional[ModelCache]) -> ModelCache:

Loading…
Cancel
Save