summaryrefslogtreecommitdiff
path: root/scripts/pa_neighbor_sets.rb
blob: 1975ffd9f6c50df7141dfabad104f0c4f86d9576 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
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