diff options
Diffstat (limited to 'scripts/test-validation-results.rb')
-rwxr-xr-x | scripts/test-validation-results.rb | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/scripts/test-validation-results.rb b/scripts/test-validation-results.rb index 2750019..7ebea89 100755 --- a/scripts/test-validation-results.rb +++ b/scripts/test-validation-results.rb @@ -5,10 +5,12 @@ include OpenTox validation = Validation::TrainTest.find File.read("data/training-test-predictions.id").chomp data = [] -puts ["SMILES","LOAEL_measured_median","LOAEL_predicted","Error","Dataset"].join(",") validation.predictions.each do |id,p| data << [Compound.find(id).smiles, p["measurements"].median, p["value"], (p["measurements"].median-p["value"]).abs,"test-prediction"] end data.sort!{|a,b| a[1] <=> b[1]} -puts data.collect{|r| r.join ","}.join("\n") +File.open(File.join("data","training-test-predictions.csv","w+")) do |f| + f.puts ["SMILES","LOAEL_measured_median","LOAEL_predicted","Error","Dataset"].join(",") + f.puts data.collect{|r| r.join ","}.join("\n") +end |