lua-ai库中的输入和输出链接

我正在使用一个名为lua-ai的机器学习库,它使用lua语言,输出应该是接近5的数字,但实际上却是0.99999999695。 我认为输出数据的输入可能有误,但我不确定。 请给我一些建议。 https://github.com/cschen1205/lua-ai

local MLPFactory=require("MLP")
local brain=MLPFactory.create(0.5)

brain:addLayer(300) --输入层 -- Xianshun:我将其更改为4...

brain:addLayer(10) --隐藏层

brain:addLayer(1) --输出层

love.math.setRandomSeed(os.time())
output = {}

input = {}

for i = 1, 1 do

 output[i] = 6

end

for j = 1, 10 do
 input[j] = {}
for i = 1,301 do
 input[j][i] = love.math.random(0,7)
 input[j][i]  = 1
end
end

for i = 1, 10 do

 brain:forwardProp(input[i])

end

for j = 1, 10 do

 output[j] = love.math.random(0,7)
 output[j] = 5
 brain:backwardProp(output)

end

for j = 1, 10 do
--output_one =  brain:getMSE(output)
 output_one = brain:forwardProp(input[j])
for i = 1, 1 do
 print( output_one[i]  )
 end

end
点赞