julia-regression-initial.jl

using Plots
using Flux
using NNlib
using Flux: @epochs

m = Dense(1,1)

loss(x, y) = sum((m(x).-y).^2)

dataset = [([0.8], [1.0]),
    ([2.0], [3.0]),
    ([2.4], [2.0]),
    ([0], [0.5]),
    ([1.5], [2]),
    ([3], [2.5]),
    ([4.0], [1.5])]

@epochs 100 Flux.train!(loss, Flux.params(m), dataset, ADAM())

x = 0:5
y =zeros(6)

for i in x
    y[i+1] = m([i])[1]
end;

plot(dataset, seriestype = :scatter, legend = false); plot!(x,y)