gistfile1.txt

require 'rubygems' 
require 'naive_bayes'


cls = NaiveBayes.new(:yes,:no,:maybe)
#put in words to ignore inside single quotes separated by commas

file=File.readlines('test.txt')

#classifies tweets from each row
file.each do |line|
     a = line.split("\t")
     d = a[0].split(' ')
     counter=0
     d.each do
     puts d[counter]
     if a[1].to_i == 0
          cls.train(:no,d[counter])
     end
     if a[1].to_i == 1 
          cls.train(:yes,d[counter])
     end
     if a[1].to_i == 2 
          cls.train(:maybe,d[counter])
     end
     counter=counter+1
     end
 end
 
 line = "Good chicken is good"
 b = line.split(' ')
 print cls.classify(b)
 
 tester = File.readlines('comp.txt')
 File.open('alcohol_classify.txt') do |s|
     file.each do |line|
          b = line.split(' ')
          s.print cls.classify(b)
     end
 end