diff --git a/analytics/models/custom_model.py b/analytics/models/custom_model.py index f1005ef..9cf4fc7 100644 --- a/analytics/models/custom_model.py +++ b/analytics/models/custom_model.py @@ -3,15 +3,8 @@ import utils import pandas as pd from typing import Optional -# Paste your model here: -class CustomModel(Model): - def __init__(self): - super() - # Use self.state to store results of your learning - # It will be saved in filesystem and loaded after server restart - self.state = {} - +class CustomModel(Model): def fit(self, dataframe: pd.DataFrame, segments: list, cache: Optional[dict]) -> dict: pass