mirror of
https://github.com/lxsang/antd-lua-plugin
synced 2024-12-29 10:48:21 +01:00
65 lines
1.8 KiB
Lua
65 lines
1.8 KiB
Lua
|
#! /usr/bin/env lua
|
||
|
|
|
||
|
|
require("lfann")
|
||
|
|
|
||
|
|
function eq(x, y)
|
||
|
|
return math.abs(x - y) < 0.00001
|
||
|
|
end
|
||
|
|
|
||
|
|
local net = fann.Net.create_shortcut{2, 1}
|
||
|
|
net:set_callback() -- Disable logging
|
||
|
|
|
||
|
|
net:set_cascade_candidate_change_fraction(0.1)
|
||
|
|
assert(eq(net:get_cascade_candidate_change_fraction(), 0.1))
|
||
|
|
|
||
|
|
net:set_cascade_candidate_stagnation_epochs(5)
|
||
|
|
assert(net:get_cascade_candidate_stagnation_epochs(5) == 5)
|
||
|
|
|
||
|
|
net:set_cascade_output_change_fraction(0.2)
|
||
|
|
assert(eq(net:get_cascade_output_change_fraction(), 0.2))
|
||
|
|
|
||
|
|
net:set_cascade_output_stagnation_epochs(10)
|
||
|
|
assert(net:get_cascade_output_stagnation_epochs(10) == 10)
|
||
|
|
|
||
|
|
net:set_cascade_weight_multiplier(0.359)
|
||
|
|
assert(eq(net:get_cascade_weight_multiplier(), 0.359))
|
||
|
|
|
||
|
|
net:set_cascade_candidate_limit(800)
|
||
|
|
assert(eq(net:get_cascade_candidate_limit(), 800))
|
||
|
|
|
||
|
|
net:set_cascade_max_cand_epochs(100)
|
||
|
|
assert(net:get_cascade_max_cand_epochs() == 100)
|
||
|
|
|
||
|
|
net:set_cascade_max_out_epochs(120)
|
||
|
|
assert(net:get_cascade_max_out_epochs() == 120)
|
||
|
|
|
||
|
|
net:set_cascade_min_out_epochs(20)
|
||
|
|
assert(net:get_cascade_min_out_epochs() == 20)
|
||
|
|
|
||
|
|
net:set_cascade_min_cand_epochs(30)
|
||
|
|
assert(net:get_cascade_min_cand_epochs() == 30)
|
||
|
|
|
||
|
|
net:set_cascade_activation_functions{fann.SIGMOID, fann.GAUSSIAN}
|
||
|
|
local funcs = net:get_cascade_activation_functions()
|
||
|
|
assert(funcs[1] == fann.SIGMOID)
|
||
|
|
assert(funcs[2] == fann.GAUSSIAN)
|
||
|
|
assert(#funcs == net:get_cascade_activation_functions_count())
|
||
|
|
|
||
|
|
net:set_cascade_num_candidate_groups(3)
|
||
|
|
assert(net:get_cascade_num_candidate_groups(), 3)
|
||
|
|
|
||
|
|
net:set_cascade_activation_steepnesses{0.5, 0.25, 0.75}
|
||
|
|
local steps = net:get_cascade_activation_steepnesses()
|
||
|
|
assert(eq(steps[1], 0.5), steps[1])
|
||
|
|
assert(eq(steps[2], 0.25))
|
||
|
|
assert(eq(steps[3], 0.75))
|
||
|
|
assert(#steps == net:get_cascade_activation_steepnesses_count())
|
||
|
|
|
||
|
|
net:cascade_train_on_file("train.data", 10, 1, 0.1)
|
||
|
|
|
||
|
|
print("End.")
|
||
|
|
|
||
|
|
if arg[1] == "mem" then
|
||
|
|
io.read("*n")
|
||
|
|
end
|