Load classifer db into sample data hash

This commit is contained in:
Joshua Peek
2012-07-23 13:13:52 -05:00
parent 97ae7c1a11
commit 0c9a947f39
5 changed files with 52 additions and 63 deletions

View File

@@ -10,8 +10,7 @@ end
file 'lib/linguist/samples.yml' => Dir['samples/**/*'] do |f|
require 'linguist/sample'
classifier = Linguist::Sample.classifier
File.open(f.name, 'w') { |io| classifier.to_yaml(io) }
File.open(f.name, 'w') { |io| Linguist::Sample.serialize_to_yaml(Linguist::Sample::DATA, io) }
end
CLOBBER.include 'lib/linguist/samples.yml'
@@ -32,7 +31,7 @@ namespace :classifier do
next if file_language.nil? || file_language == 'Text'
begin
data = open(file_url).read
guessed_language, score = Linguist::Classifier.instance.classify(data).first
guessed_language, score = Linguist::Classifier.new(Sample::DATA).classify(data).first
total += 1
guessed_language == file_language ? correct += 1 : incorrect += 1