blob: 666ebdb847e7c4027d5dc21f848f10c4a7751c7a (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
|
require 'torch'
require 'optim'
require 'rosenbrock'
require 'l2'
-- 10-D rosenbrock
x = torch.Tensor(2):fill(0)
config = {popsize=50, scaleFactor=0.5, crossoverRate=0.9, maxFEs=3000}
-- will take some time
x,fx=optim.de(rosenbrock,x,config)
print('Rosenbrock test')
print()
-- approx 6500 function evals expected
print('Number of function evals = ',i)
print('x=');print(x)
print('fx=')
for i=1,config.maxFEs do print(i,fx[i]); end
print()
print()
|
