首页 > 解决方案 > 在 lua-ai 库中链接输入和输出

问题描述

我使用的是lua语言中一个叫lua-ai的机器学习库,输出应该是接近5的数字,结果却是0.99999999695。我认为输出数据的输入可能是错误的,但我不知道。请给我一些建议。 https://github.com/cschen1205/lua-ai

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

brain:addLayer(300) --input layer --Xianshun: I change it to 4...

brain:addLayer(10) --hidden layer

brain:addLayer(1) --output layer

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

标签: machine-learninglua

解决方案


推荐阅读