Trying to get started with Flux.jl. I’m trying estimate a linear regression. I have the following: using Flux using Random Random.seed!(1234) β = [10., 20., 30.] # true parameters predict(x) = (x*β) function loss(x, y) ŷ = predict(x) sum((ŷ .- y).^2) end x = hcat(rand(500,2), ones(500,…

Note that Flux.train! only trains for one “epoch”. In this case, that means one iteration of gradient descent with the default step size. The training section of the documentation elaborates on how to train your model for many “epochs”. In general you will have to implement your own convergence cri…

Yep, that was it. Coming from another discipline, I thought “training” meant estimation. It’s very strange to me to have the training stop after one iteration…

Yeah, it is somewhat confusing nomenclature. In general, it’s something of a general design philosophy of Flux to provide basic tools and assume very little about what you want to do with them. If you look at the source code of Flux.train! , for example, besides the error checking it’s really just a …

I do something similar here: [image] Generic Function to train NN w/ Flux Machine Learning Updated: I would like to write a generic function to train a model in Flux. An ordinary linear regression w/ intercept is a special case of a neural network. If the optimizers work…

that’s very useful, thank you!

Regression in Flux.jl

Specific Domains Machine Learning

Albert_Zevelev May 4, 2020, 5:01pm 5

I do something similar here:

1 Like

Topic		Replies	Views
Generic Function to train NN w/ Flux Machine Learning flux	7	1688	April 14, 2020
Flux function fitting Machine Learning flux	2	1088	August 7, 2020
No changes with Flux NN regression training Machine Learning question	2	678	October 2, 2020
Training a simple linear model in Flux Machine Learning	4	2553	September 11, 2020
Params not getting updated during training New to Julia flux	25	1897	October 11, 2020

Regression in Flux.jl

Related topics