db_clustering 0.1.9 → 0.1.10

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: cd346d3f3511320811d7fb97312e590947728903
-  data.tar.gz: 1f9a47c7dfbdffb8ecba30c0c8b4e7ec30f1a895
+  metadata.gz: f9174e0f67fcea91843ba237b2cbb1e214558c8f
+  data.tar.gz: 01b85e27ed68fc16581c23324a155064d3b6139f
 SHA512:
-  metadata.gz: bb79a9eba3cecb5a72b1156df2d189f03bc91ae73d620bd3a2b289a9d596fc2e56325e1f1b81c3dfdabb062c6186a36df637dc6f3dd30dde715d775d48b37873
-  data.tar.gz: a5f2065ad7a8c8d1d28cb29d90c33955d6fd6c69397cfd4f456ad3ac20580e0d011cb4958dd56030aa5aaa1f1149b5a9f9f36d8c97e4131234329ffa32c3073c
+  metadata.gz: 5fec9b5ba8dcbb1f4f0ad6668ffcb752bbdd5b691b4263608d0854a7b4036f5a223396539d4fbb7a36c4c085067a47fb860c9701ff580e5d2e81b7f964bf276d
+  data.tar.gz: 0fae1b29b2f0feece92f5cc9286eb3376d89cbb9bec71ef94e66a897dce7c554449c2dde60152a6d2334e34d0708559eb0777c8b4d85930c68c7330dc421886d

data/VERSION

@@ -1 +1 @@
-0.1.9
+0.1.10

data/lib/algorithms/density_based/dbscan.rb

@@ -21,25 +21,29 @@ module DbClustering
 
           if neighbors.count < min_neighbors
             point.is_noise = true
-          else
-            if point.cluster.nil?
-              cluster = DbClustering::Models::Cluster.new
-              @clusters << cluster
-            else
-              cluster = point.cluster
-            end
-
-            neighbors.each do |neighbor|
-              if neighbor.cluster.nil?
-                cluster.add(neighbor)
-              end
-            end
+          elsif point.cluster.nil?
+            cluster = DbClustering::Models::Cluster.new
+            @clusters << cluster
+            cluster.add(point)
+
+            expand_cluster(point, neighbors, max_distance)
           end
 
           yield(point, current_index, points_count) if block_given?
         end
       end
 
+      def expand_cluster(point, neighbors, max_distance)
+        neighbors.each do |neighbor|
+          if neighbor.cluster.nil?
+            point.cluster.add(neighbor)
+
+            neighbors = @datasource.neighbors(point: point, distance_metric: @distance_metric, max_distance: max_distance)
+            expand_cluster(neighbor, neighbors, max_distance)
+          end
+        end
+      end
+
     end
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: db_clustering
 version: !ruby/object:Gem::Version
-  version: 0.1.9
+  version: 0.1.10
 platform: ruby
 authors:
 - Cihat Gündüz