unique neighbor sets

1 files changed, 16 insertions, 0 deletions

diff --git a/scripts/pa_neighbor_sets.rb b/scripts/pa_neighbor_sets.rb
new file mode 100755
index 0000000..1975ffd
--- /dev/null
+++ b/scripts/pa_neighbor_sets.rb
@@ -0,0 +1,16 @@
+#!/usr/bin/env ruby
+require_relative '../../lazar/lib/lazar'
+include OpenTox
+
+results = JSON.parse File.read(ARGV[0])
+neighbors = []
+results.each do |s,r|
+  neighbors << r["neighbors"] if r["value"]
+end
+
+o = {
+  :nr_predictions => neighbors.size,
+  :unique_neighbor_sets => neighbors.uniq.size,
+  :unique_neighbor_sets_without_similarity => neighbors.collect{|s| s.collect{|n| n["id"]}.sort}.uniq.size
+}
+puts o.to_json