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...

Garanta sua matrícula hoje e ganhe + 2 meses grátis

Continue sua jornada tech com ainda mais tempo para aprender e evoluir

Quero aproveitar agora
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)