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.

20 lines
524 B

6 years ago
from models import Model
import utils
import pandas as pd
from typing import Optional
6 years ago
# 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 = {}
def fit(self, dataframe: pd.DataFrame, segments: list, cache: Optional[dict]) -> dict:
6 years ago
pass
def predict(self, dataframe, cache: Optional[dict]):
6 years ago
return []