Скользящая линейная регрессия для использования с операцией groupby на фрейме данных cuDF - PullRequest
Новая
       68

Скользящая линейная регрессия для использования с операцией groupby на фрейме данных cuDF

2 голосов
/ 02 августа 2020
• 1000 * 
def RollingOLS(x, y):
    lr = LinearRegression(fit_intercept = True, normalize = False, algorithm = 'svd')
    reg = lr.fit(x, y)
    
    return reg.coef_

Что бы я хотел сделать: 

data.groupby('date').rolling(2).apply(RollingOLS, x=x_value, y=y_value)

Однако я получаю сообщение об ошибке: NotImplementedError: Handling UDF with null values is not yet supported. Есть ли способ побороть эту ошибку? Спасибо.

1 Ответ

1 голос
/ 03 августа 2020

Чтобы устранить ошибку NotImplementedError: Handling UDF with null values is not yet supported, переназначьте значения None/null другому значению или удалите образцы, которые имеют значения None/null из вашего DataFrame.

...