summaryrefslogtreecommitdiff
path: root/scripts/test-validation-results.rb
blob: b119bb734d13976aa6f7e7717daef0175f851c8b (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
#!/usr/bin/env ruby
require_relative '../../lazar/lib/lazar'
include OpenTox

validation = Validation::TrainTest.find File.read("data/training-test-predictions.id").chomp

data = []
validation.predictions.each do |id,p|
  warnings = "F"
  warnings = "T" if p["warnings"] and !p["warnings"].empty? 
  data << [Compound.find(id).smiles, p["measurements"].median, p["value"], (p["measurements"].median-p["value"]).abs,"test-prediction",warnings]
end

data.sort!{|a,b| a[1] <=> b[1]}
File.open(File.join("data","training-test-predictions.csv"),"w+") do |f|
  f.puts ["SMILES","LOAEL_measured_median","LOAEL_predicted","Error","Dataset","Warnings"].join(",")
  f.puts data.collect{|r| r.join ","}.join("\n")
end