db_clustering 0.1.10 → 0.1.11

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: f9174e0f67fcea91843ba237b2cbb1e214558c8f
-  data.tar.gz: 01b85e27ed68fc16581c23324a155064d3b6139f
+  metadata.gz: f960d170fbc8d820ad18016a4ce03943fd02de54
+  data.tar.gz: c257769487f5a8efaa1d2a2606aa1aee51083453
 SHA512:
-  metadata.gz: 5fec9b5ba8dcbb1f4f0ad6668ffcb752bbdd5b691b4263608d0854a7b4036f5a223396539d4fbb7a36c4c085067a47fb860c9701ff580e5d2e81b7f964bf276d
-  data.tar.gz: 0fae1b29b2f0feece92f5cc9286eb3376d89cbb9bec71ef94e66a897dce7c554449c2dde60152a6d2334e34d0708559eb0777c8b4d85930c68c7330dc421886d
+  metadata.gz: 6a02139c030777fdcd4ce942fad981682e50f641d5e99e76fd40859a77caa73199869f301b3bb272eae647f72b5a1c0cc967f4a9a535b5e9b229b8d1e25a246a
+  data.tar.gz: 972a4d5b380333a128901d62dc7b28bf8d4f5240f3517e55eb0404e99ccb7e5164299edfb5a55a87e917d58608c45d15896c236caf56c993bbc161a94db65ab0

data/VERSION

@@ -1 +1 @@
-0.1.10
+0.1.11

data/lib/algorithms/density_based/dbscan.rb

@@ -21,29 +21,28 @@ module DbClustering
 
           if neighbors.count < min_neighbors
             point.is_noise = true
-          elsif point.cluster.nil?
-            cluster = DbClustering::Models::Cluster.new
-            @clusters << cluster
-            cluster.add(point)
-
-            expand_cluster(point, neighbors, max_distance)
+          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)
+
+                # add the neighbors of the neighbor to the neighbors to fully expand the cluster
+                neighbors |= @datasource.neighbors(point: neighbor, distance_metric: @distance_metric, max_distance: max_distance)
+              end
+            end
           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.10
+  version: 0.1.11
 platform: ruby
 authors:
 - Cihat Gündüz