INNER CODE UNIT · Python

Model

dmarienko/QuantTradingResearch · models/helpers.py:39

class Model:
    """
    Abstract class for any tracking models
    """
    def __init__(self, description=None):
        self.description = description
    
    def fit(self, x, y, **kwargs):
        return self
    
    def predict(self, x, y=None, **kwargs):
        return None
    

class TrackingModel:
    def __init__(self, data, index_name, train_date):
        self.data = data
        self.closes = retain_columns_and_join(data, 'close')

View source record →