Regression in Flux.jl

[Albert_Zevelev]

I do something similar here:

Generic Function to train NN w/ Flux

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 well I should get the same result (esp from such a simple model). Unfortunately I don’t. First: prepare Boston housing data using MLJ: @load_boston, partition using Flux, Statistics X, y = @load_boston; X = hcat(X...); train, test = partition(eachindex(y), .7, rng=333); # Xm = hcat(X, size(X,1) |> ones ); #add …