blob: ae72b8e4ba74713557bfb4271eaa96d2d7f0a33d (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
|
#!/usr/bin/env ruby
require 'csv'
tp = 0
fp = 0
tn = 0
fn = 0
pred = CSV.read(ARGV[0],headers: true,:col_sep => ",")
act = CSV.read(File.join("data","training","mutagenicity.csv"),headers: true,:col_sep => ",")
data = {}
pred.each do |row|
row[1].to_f < 0.5 ? p = 0 : p = 1
data[row[0]] =[p]
end
act.each do |row|
data[row[0]] << row[1].to_i if data[row[0]]
end
data.each do |smi,a|
tp += 1 if a[0] == 1 and a[1] == 1
tn += 1 if a[0] == 0 and a[1] == 0
fp += 1 if a[0] == 0 and a[1] == 1
fn += 1 if a[0] == 1 and a[1] == 0
end
puts "#{tp},#{fp}\n#{fn},#{tn}"
|