#!/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