require 'rubygems'
require 'classifier'


classifier = if File.exists?('classifier.data')
 File.open('classifier.data') do|file|
   Marshal.load(file)
 end
else
 Classifier::Bayes.new "ruby"
end


path_or_file = "test-file.rb"

lang = "ruby"

classifier.train("ruby", File.open(path_or_file).read)

File.open('classifier.data','w') do|file|
 Marshal.dump(classifier, file)
end