INNER CODE UNIT · Python

fit

dmarienko/QuantTradingResearch · models/helpers.py:46

    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')
        self.index = data[index_name]
        self.index_closes = self.index.close
        self.X_price = self.closes[self.closes.columns[~self.closes.columns.str.match(index_name)]]
        self.Y_price = self.closes[index_name]
        self.X_ret, self.Y_ret = prices_to_returns(self.X_price), prices_to_returns(self.Y_price)
        self.train_date = pd.Timestamp(train_date)
        self.selection = None

View source record →