Hi everyone, I’m new to Julia and I’m trying to create a loop for a regression model. For example, list = [:A, :B, :C] for i in list result = lm(@formula(i ~ X1 + X2), data) println(result) end Something like this. But I get an error. type NamedTuple has no field i Can someone teach …

The recommended way of constructing formulas programmatically is to use Terms (or the term function) like so: [lm(Term(i) ~ Term(:X1) + Term(:X2), data) for i in list] this is discussed in the docs here .

I could solve it by not using the macro “@formula”. Instead, list = [:A, :B, :C] for i in list result = lm(StatsModels.Term(i) ~ StatsModels.Term(:X1) + StatsModels.Term(:X2), data) println(result) end Please let me know if you have any other solution to this. Thanks!

In the expression @formula(i ~ X1 + X2) you have to specify that i is not a column name in your data, but that it contains a column name… You can do it like this: for i in list result = lm((@eval @formula($i ~ X1 + X2)), data) println(result) end

Thank you! It works. Do you also have any idea of how to save the results for each regression? For example, result1, result2, …

You can collect them using array comprehension: using DataFrames, GLM data = DataFrame(X1=[1,2,3,4], X2=[2,4,7,8], A=[4,8,13,17], B=[10,11,12,13]); list = [:A, :B]; results = [lm((@eval @formula($i ~ X1 + X2)), data) for i in list] This gives you results in results[1], results[2], …

@nilshg Thank you! :slight_smile:

How to create a loop for a regression model?

Specific Domains Statistics

Topic		Replies	Views
How to use lm inside a for loop for different variables Statistics	2	683	June 8, 2022
Using all independent variables with @formula in a multiple linear model New to Julia glm	18	4611	January 29, 2023
Univariate linear regression with each covariate of dataframe Statistics dataframes , regression	5	846	October 15, 2019
Iteration over 10 columns to calculate LinearRegression Machine Learning	4	421	March 27, 2021
Using all variables on a regression? General Usage	0	494	November 20, 2018

How to create a loop for a regression model?

Related topics