mirror of
https://github.com/KevinMidboe/linguist.git
synced 2025-10-29 17:50:22 +00:00
Sort classifier yaml output
This commit is contained in:
@@ -144,6 +144,39 @@ module Linguist
|
|||||||
def language_probability(language)
|
def language_probability(language)
|
||||||
Math.log(@languages[language].to_f / @languages_total.to_f)
|
Math.log(@languages[language].to_f / @languages_total.to_f)
|
||||||
end
|
end
|
||||||
|
|
||||||
|
# Public: Serialize classifier to YAML.
|
||||||
|
#
|
||||||
|
# opts - Hash of YAML options.
|
||||||
|
#
|
||||||
|
# Returns nothing.
|
||||||
|
def to_yaml(io)
|
||||||
|
data = "--- !ruby/object:Linguist::Classifier\n"
|
||||||
|
|
||||||
|
data << "languages_total: #{@languages_total}\n"
|
||||||
|
data << "tokens_total: #{@tokens_total}\n"
|
||||||
|
|
||||||
|
data << "languages:\n"
|
||||||
|
@languages.sort.each do |language, count|
|
||||||
|
data << " #{language.inspect}: #{count}\n"
|
||||||
|
end
|
||||||
|
|
||||||
|
data << "language_tokens:\n"
|
||||||
|
@language_tokens.sort.each do |language, count|
|
||||||
|
data << " #{language.inspect}: #{count}\n"
|
||||||
|
end
|
||||||
|
|
||||||
|
data << "tokens:\n"
|
||||||
|
@tokens.sort.each do |language, tokens|
|
||||||
|
data << " #{language.inspect}:\n"
|
||||||
|
tokens.sort.each do |token, count|
|
||||||
|
data << " #{token.inspect}: #{count}\n"
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
io.write data
|
||||||
|
nil
|
||||||
|
end
|
||||||
end
|
end
|
||||||
|
|
||||||
# Eager load instance
|
# Eager load instance
|
||||||
|
|||||||
Reference in New Issue
Block a user