models/mutagenicity/download/hansen_convert.rb


1
2
3
4
5
6
7
8
9
10
11
12

#!/usr/bin/env ruby
require 'csv'

# convert hansen dataset
hansen = CSV.read ARGV[0]
hansen.shift

map = {"0" => "non-mutagenic","1" => "mutagenic"}
puts "ID,SMILES,Mutagenicity"
hansen.each do |row|
  puts [row[0],row[5],map[row[2]]].join "," 
end