1
resposta

Aplicar previsão em cada linha do DF

Olá, estou no meio do curso de Regressão Linear que estou tentando aplicar o modelo de regressão em cada linha do DF, porém não estou conseguindo. Acredito que o loop for possa ajudar, mas confesso que não entendi como fazer.

A minha lógica em aplicar o for seria a seguinte, vejo quantas linhas tem DF e vou aplicando até zerar as linhas...

1 resposta

Se entendi certo o questionamento (muito com base no titulo da tópico), você pode usar a função .predict() junto com o modelo e os dados, ele fará a previsão para cada entrada dos dados.

Ex:

model = OLS(data)
ols_fit = model.fit()
prediction = ols_fit.predict(data)