ant_colony_tsp 0.0.1

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
checksums.yaml ADDED
@@ -0,0 +1,15 @@
1
+ ---
2
+ !binary "U0hBMQ==":
3
+ metadata.gz: !binary |-
4
+ ODc3Mjg0M2ZkMTBjMzVhM2MzZGY4MjBhMTg0ZWRkMmNkNjE1OTk0NA==
5
+ data.tar.gz: !binary |-
6
+ YTY0ZjkxYzNkZTA3MTJmODhlNjY3NTAxM2FjYWY5ZGJhYTllYTNmYw==
7
+ SHA512:
8
+ metadata.gz: !binary |-
9
+ ZTM0ZjA5ODkzMDY0MTkwNDY2NTcwNjlhMTQxODhmZWY0NDFhMDM0YWY3ZGIx
10
+ YzI5OGE4ZGFmZGU4OTUxYzY2OTM2OTBjZWVmZWM2MGUyZjFkY2IyNmU3Y2Ux
11
+ Mzk3NzAxMDJhYTg2YzZjZGJiYzQyYTQxZDg1NmNhODYxOTc1YTU=
12
+ data.tar.gz: !binary |-
13
+ M2M4OThjYjUxOGFhNTI2YjVhODc1ODQ1YjNiZGQwOWI0Nzg1ZmI3YzU3YTI4
14
+ ZGQ1ZjJhOTZkYmI0M2ViOGZlYTMzOTI3NmRjNzJiYmYyNGRlNDUwNmFjYmZm
15
+ N2ZkZTFkOWMzYWY3YTUwZmEzNjExYmVjNjI0NmY4MjkxZmZjMWY=
data/.gitignore ADDED
@@ -0,0 +1,17 @@
1
+ *.gem
2
+ *.rbc
3
+ .bundle
4
+ .config
5
+ .yardoc
6
+ Gemfile.lock
7
+ InstalledFiles
8
+ _yardoc
9
+ coverage
10
+ doc/
11
+ lib/bundler/man
12
+ pkg
13
+ rdoc
14
+ spec/reports
15
+ test/tmp
16
+ test/version_tmp
17
+ tmp
data/Gemfile ADDED
@@ -0,0 +1,4 @@
1
+ source 'https://rubygems.org'
2
+
3
+ # Specify your gem's dependencies in ant_colony_tsp.gemspec
4
+ gemspec
data/LICENSE.txt ADDED
@@ -0,0 +1,22 @@
1
+ Copyright (c) 2013 dusan
2
+
3
+ MIT License
4
+
5
+ Permission is hereby granted, free of charge, to any person obtaining
6
+ a copy of this software and associated documentation files (the
7
+ "Software"), to deal in the Software without restriction, including
8
+ without limitation the rights to use, copy, modify, merge, publish,
9
+ distribute, sublicense, and/or sell copies of the Software, and to
10
+ permit persons to whom the Software is furnished to do so, subject to
11
+ the following conditions:
12
+
13
+ The above copyright notice and this permission notice shall be
14
+ included in all copies or substantial portions of the Software.
15
+
16
+ THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
17
+ EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
18
+ MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
19
+ NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
20
+ LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
21
+ OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
22
+ WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
data/README.md ADDED
@@ -0,0 +1,29 @@
1
+ # AntColonyTsp
2
+
3
+ Ant colony to solve travelling salesman
4
+
5
+ ## Installation
6
+
7
+ Add this line to your application's Gemfile:
8
+
9
+ gem 'ant_colony_tsp'
10
+
11
+ And then execute:
12
+
13
+ $ bundle
14
+
15
+ Or install it yourself as:
16
+
17
+ $ gem install ant_colony_tsp
18
+
19
+ ## Usage
20
+
21
+ TODO: Write usage instructions here
22
+
23
+ ## Contributing
24
+
25
+ 1. Fork it
26
+ 2. Create your feature branch (`git checkout -b my-new-feature`)
27
+ 3. Commit your changes (`git commit -am 'Add some feature'`)
28
+ 4. Push to the branch (`git push origin my-new-feature`)
29
+ 5. Create new Pull Request
data/Rakefile ADDED
@@ -0,0 +1,6 @@
1
+ require "bundler/gem_tasks"
2
+ require "rspec/core/rake_task"
3
+
4
+ RSpec::Core::RakeTask.new("spec")
5
+
6
+ task :default => :spec
@@ -0,0 +1,24 @@
1
+ # coding: utf-8
2
+ lib = File.expand_path('../lib', __FILE__)
3
+ $LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
4
+ require 'ant_colony_tsp/version'
5
+
6
+ Gem::Specification.new do |spec|
7
+ spec.name = "ant_colony_tsp"
8
+ spec.version = AntColonyTsp::VERSION
9
+ spec.authors = ["no-glue"]
10
+ spec.email = ["nikolapav1985@gmail.com"]
11
+ spec.description = %q{ant colony to solve travelling salesman}
12
+ spec.summary = %q{ant colony to solve travelling salesman}
13
+ spec.homepage = ""
14
+ spec.license = "MIT"
15
+
16
+ spec.files = `git ls-files`.split($/)
17
+ spec.executables = spec.files.grep(%r{^bin/}) { |f| File.basename(f) }
18
+ spec.test_files = spec.files.grep(%r{^(test|spec|features)/})
19
+ spec.require_paths = ["lib"]
20
+
21
+ spec.add_development_dependency "bundler", "~> 1.3"
22
+ spec.add_development_dependency "rake"
23
+ spec.add_development_dependency "rspec"
24
+ end
@@ -0,0 +1,120 @@
1
+ require "ant_colony_tsp/version"
2
+
3
+ module AntColonyTsp
4
+ class AntColonyTsp
5
+ # distance, between cities
6
+ def euc_2d(c1, c2)
7
+ Math.sqrt((c1[0] - c2[0]) ** 2.0 + (c1[1] - c2[1]) ** 2.0).round
8
+ end
9
+
10
+ # cost, of the tour
11
+ def cost(sh, cities)
12
+ distance = 0
13
+ sh.each_with_index do |c1, i|
14
+ c2 = (i == (sh.size - 1)) ? sh[0] : sh[i + 1]
15
+ distance += euc_2d(cities[c1], cities[c2])
16
+ end
17
+ distance
18
+ end
19
+
20
+ # shake, cities
21
+ def shake(cities)
22
+ sh = Array.new(cities.size){|i| i}
23
+ sh.each_index do |i|
24
+ r = rand(sh.size - i) + i
25
+ sh[r], sh[i] = sh[i], sh[r]
26
+ end
27
+ sh
28
+ end
29
+
30
+ # phero matrix, get it
31
+ def get_phero_matrix(num_cities, init_pher)
32
+ return Array.new(num_cities){|i| Array.new(num_cities, init_pher)}
33
+ end
34
+
35
+ # where to go next
36
+ def get_choices(cities, last_city, skip, pheromone, c_heur, c_hist)
37
+ choices = []
38
+ cities.each_with_index do |position, i|
39
+ next if skip.include?(i)
40
+ prob = {:city => i}
41
+ prob[:history] = pheromone[last_city][i] ** c_hist
42
+ prob[:distance] = euc_2d(cities[last_city], position)
43
+ prob[:heuristic] = (1.0 / prob[:distance]) ** c_heur
44
+ prob[:prob] = prob[:history] * prob[:heuristic]
45
+ choices << prob
46
+ end
47
+ choices
48
+ end
49
+
50
+ # select city, next
51
+ def select_city(choices)
52
+ sum = choices.inject(0.0) {|sum, item| sum + item[:prob]}
53
+ return choices[rand(choices.size)][:city] if sum == 0.0
54
+ v = rand()
55
+ choices.each_with_index do |choice, i|
56
+ v -= (choice[:prob] / sum)
57
+ return choice[:city] if v <= 0.0
58
+ end
59
+ return choices.last[:city]
60
+ end
61
+
62
+ # select city, greedy
63
+ def select_city_greedy(choices)
64
+ return choices.max{|a, b| a[:prob] <=> b[:prob]}[:city]
65
+ end
66
+
67
+ # get tour +++
68
+ def get_tour(cities, phero, c_heur, c_greed)
69
+ sh = []
70
+ sh << rand(cities.size)
71
+ begin
72
+ choices = get_choices(cities, sh.last, sh, phero, c_heur, 1.0)
73
+ greedy = rand() <= c_greed
74
+ next_city = (greedy) ? select_city_greedy(choices) : select_city(choices)
75
+ sh << next_city
76
+ end until sh.size == cities.size
77
+ sh
78
+ end
79
+
80
+ # update phero, global
81
+ def update_phero_global(phero, candidate, decay)
82
+ candidate[:vector].each_with_index do |row, i|
83
+ col = (i == candidate[:vector].size - 1) ? candidate[:vector][0] : candidate[:vector][i + 1]
84
+ value = ((1 - decay) * phero[row][col]) + (decay * (1 / candidate[:cost]))
85
+ phero[row][col] = value
86
+ phero[col][row] = value
87
+ end
88
+ end
89
+
90
+ # update phero, local
91
+ def update_phero_local(phero, candidate, c_local_phero, init_phero)
92
+ candidate[:vector].each_with_index do |row, i|
93
+ col = (i == candidate[:vector].size - 1) ? candidate[:vector][0] : candidate[:vector][i + 1]
94
+ value = (1.0 - c_local_phero) * phero[row][col] + (c_local_phero * init_phero)
95
+ phero[row][col] - value
96
+ phero[col][row] = value
97
+ end
98
+ end
99
+
100
+ # search
101
+ def search(cities, max_it, num_ants, decay, c_heur, c_local_phero, c_greed)
102
+ best = {:vector => shake(cities)}
103
+ best[:cost] = cost(best[:vector], cities)
104
+ init_phero = 1.0 / (cities.size.to_f * best[:cost])
105
+ phero = get_phero_matrix(cities.size, init_phero)
106
+ max_it.times do |iter|
107
+ num_ants.times do
108
+ cand = {}
109
+ cand[:vector] = get_tour(cities, phero, c_heur, c_greed)
110
+ cand[:cost] = cost(cand[:vector], cities)
111
+ best = cand if cand[:cost] < best[:cost]
112
+ update_phero_local(phero, cand, c_local_phero, init_phero)
113
+ end
114
+ update_phero_global(phero, best, decay)
115
+ puts " > iteration #{iter + 1}, best #{best[:cost]}"
116
+ end
117
+ best
118
+ end
119
+ end
120
+ end
@@ -0,0 +1,3 @@
1
+ module AntColonyTsp
2
+ VERSION = "0.0.1"
3
+ end
@@ -0,0 +1,15 @@
1
+ require 'spec_helper'
2
+
3
+ describe AntColonyTsp do
4
+ it 'gets 7542 or greater for berlin' do
5
+ expect(AntColonyTsp::AntColonyTsp.new.search([[565,575],[25,185],[345,750],[945,685],[845,655],
6
+ [880,660],[25,230],[525,1000],[580,1175],[650,1130],[1605,620],
7
+ [1220,580],[1465,200],[1530,5],[845,680],[725,370],[145,665],
8
+ [415,635],[510,875],[560,365],[300,465],[520,585],[480,415],
9
+ [835,625],[975,580],[1215,245],[1320,315],[1250,400],[660,180],
10
+ [410,250],[420,555],[575,665],[1150,1160],[700,580],[685,595],
11
+ [685,610],[770,610],[795,645],[720,635],[760,650],[475,960],
12
+ [95,260],[875,920],[700,500],[555,815],[830,485],[1170,65],
13
+ [830,610],[605,625],[595,360],[1340,725],[1740,245]], 100, 10, 0.1, 2.5, 0.1, 0.9)[:cost]).to be >= 7542
14
+ end
15
+ end
@@ -0,0 +1,6 @@
1
+ require 'rubygems'
2
+ require 'bundler/setup'
3
+ require 'ant_colony_tsp'
4
+
5
+ RSpec.configure do |config|
6
+ end
metadata ADDED
@@ -0,0 +1,98 @@
1
+ --- !ruby/object:Gem::Specification
2
+ name: ant_colony_tsp
3
+ version: !ruby/object:Gem::Version
4
+ version: 0.0.1
5
+ platform: ruby
6
+ authors:
7
+ - no-glue
8
+ autorequire:
9
+ bindir: bin
10
+ cert_chain: []
11
+ date: 2013-12-24 00:00:00.000000000 Z
12
+ dependencies:
13
+ - !ruby/object:Gem::Dependency
14
+ name: bundler
15
+ requirement: !ruby/object:Gem::Requirement
16
+ requirements:
17
+ - - ~>
18
+ - !ruby/object:Gem::Version
19
+ version: '1.3'
20
+ type: :development
21
+ prerelease: false
22
+ version_requirements: !ruby/object:Gem::Requirement
23
+ requirements:
24
+ - - ~>
25
+ - !ruby/object:Gem::Version
26
+ version: '1.3'
27
+ - !ruby/object:Gem::Dependency
28
+ name: rake
29
+ requirement: !ruby/object:Gem::Requirement
30
+ requirements:
31
+ - - ! '>='
32
+ - !ruby/object:Gem::Version
33
+ version: '0'
34
+ type: :development
35
+ prerelease: false
36
+ version_requirements: !ruby/object:Gem::Requirement
37
+ requirements:
38
+ - - ! '>='
39
+ - !ruby/object:Gem::Version
40
+ version: '0'
41
+ - !ruby/object:Gem::Dependency
42
+ name: rspec
43
+ requirement: !ruby/object:Gem::Requirement
44
+ requirements:
45
+ - - ! '>='
46
+ - !ruby/object:Gem::Version
47
+ version: '0'
48
+ type: :development
49
+ prerelease: false
50
+ version_requirements: !ruby/object:Gem::Requirement
51
+ requirements:
52
+ - - ! '>='
53
+ - !ruby/object:Gem::Version
54
+ version: '0'
55
+ description: ant colony to solve travelling salesman
56
+ email:
57
+ - nikolapav1985@gmail.com
58
+ executables: []
59
+ extensions: []
60
+ extra_rdoc_files: []
61
+ files:
62
+ - .gitignore
63
+ - Gemfile
64
+ - LICENSE.txt
65
+ - README.md
66
+ - Rakefile
67
+ - ant_colony_tsp.gemspec
68
+ - lib/ant_colony_tsp.rb
69
+ - lib/ant_colony_tsp/version.rb
70
+ - spec/ant_colony_tsp_spec.rb
71
+ - spec/spec_helper.rb
72
+ homepage: ''
73
+ licenses:
74
+ - MIT
75
+ metadata: {}
76
+ post_install_message:
77
+ rdoc_options: []
78
+ require_paths:
79
+ - lib
80
+ required_ruby_version: !ruby/object:Gem::Requirement
81
+ requirements:
82
+ - - ! '>='
83
+ - !ruby/object:Gem::Version
84
+ version: '0'
85
+ required_rubygems_version: !ruby/object:Gem::Requirement
86
+ requirements:
87
+ - - ! '>='
88
+ - !ruby/object:Gem::Version
89
+ version: '0'
90
+ requirements: []
91
+ rubyforge_project:
92
+ rubygems_version: 2.1.11
93
+ signing_key:
94
+ specification_version: 4
95
+ summary: ant colony to solve travelling salesman
96
+ test_files:
97
+ - spec/ant_colony_tsp_spec.rb
98
+ - spec/spec_helper.rb