mirror of
https://github.com/KevinMidboe/linguist.git
synced 2025-10-29 17:50:22 +00:00
Load classifer db into sample data hash
This commit is contained in:
5
Rakefile
5
Rakefile
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user