在julia中用flux.jl实现线性回归

using Flux
using Flux: @epochs, mse

# 输入特征
X = [1.0 2.0 3.0 4.0 5.0]

# 目标变量
y = [2.0 3.0 4.0 5.0 6.0]

# 定义模型
model = Chain(Dense(1, 1))

# 定义损失函数
loss(x, y) = mse(model(x), y)

# 定义优化器
optimizer = Descent(0.1)

# 迭代训练模型
@epochs 1000 Flux.train!(loss, params(model), [(X, y)], optimizer)

# 预测
prediction = model(X)