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…

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	1656	April 14, 2020
Flux function fitting Machine Learning flux	2	1057	August 7, 2020
No changes with Flux NN regression training Machine Learning question	2	664	October 2, 2020
Training a simple linear model in Flux Machine Learning	4	2515	September 11, 2020
How to use Flux for a general non-linear minimization Machine Learning flux	1	681	July 1, 2019

Regression in Flux.jl

Related topics