mirror of
https://github.com/KevinMidboe/linguist.git
synced 2025-10-29 17:50:22 +00:00
Use classifier as primary method for disambiguation
This commit is contained in:
@@ -454,30 +454,18 @@ module Linguist
|
||||
# Returns a Language or nil.
|
||||
def disambiguate_extension_language
|
||||
if Language.ambiguous?(extname)
|
||||
# name = "guess_#{extname.sub(/^\./, '')}_language"
|
||||
# send(name) if respond_to?(name)
|
||||
|
||||
possible_languages = Language.all.select { |l| l.extensions.include?(extname) }
|
||||
|
||||
name = "guess_#{extname.sub(/^\./, '')}_language"
|
||||
language = send(name) if respond_to?(name)
|
||||
|
||||
if possible_languages.any?
|
||||
results = Classifier.instance.classify(data, possible_languages)
|
||||
guessed_language, score = results.first
|
||||
|
||||
if guessed_language != language
|
||||
report_classifier_incorrect_guess(language, guessed_language, score)
|
||||
if result = Classifier.instance.classify(data, possible_languages).first
|
||||
result[0]
|
||||
end
|
||||
end
|
||||
|
||||
language
|
||||
end
|
||||
end
|
||||
|
||||
class LanguageClassifierError < StandardError; end
|
||||
|
||||
def report_classifier_incorrect_guess(expected, actual, score)
|
||||
raise LanguageClassifierError, "Expected #{expected}, but was #{actual} scoring #{score}.\n#{name}\n#{data}"
|
||||
end
|
||||
|
||||
# Internal: Guess language of .cls files
|
||||
#
|
||||
# Returns a Language.
|
||||
|
||||
Reference in New Issue
Block a user