Hi I've seen that the classifier has the fast-stemmer as a dependency right? I th

<a class="user-mention notranslate" data-hovercard-type="user" data-hovercard-url="/us

Question regarding languages about classifier-reborn HOT 3 CLOSED

jekyll commented on June 9, 2024

Question regarding languages

from classifier-reborn.

Comments (3)

Ch4s3 commented on June 9, 2024

@janorivera If you look at our stop words, especially here you'll see that we do have Spanish stop words.

Look at how the classifier is initialized, English, 'en' is the default.

def initialize(*args)
      @categories = Hash.new
      options = { language: 'en', auto_categorize: false }
      args.flatten.each do |arg|
        if arg.kind_of?(Hash)
          options.merge!(arg)
        else
          add_category(arg)
        end
      end
      @total_words = 0
      @category_counts = Hash.new(0)
      @category_word_count = Hash.new(0)
      @language = options[:language]
      @auto_categorize = options[:auto_categorize]
end

Try something like the following:

> classifier = ClassifierReborn::Bayes.new(language: 'es', 'bueno', 'malo')
SyntaxError: unexpected ')', expecting end-of-input
> classifier = ClassifierReborn::Bayes.new('bueno', 'malo', language: 'es')
=> #<ClassifierReborn::Bayes:0x007f85358ca200 @categories={:Bueno=>{}, :Malo=>{}}, @total_words=0, @category_counts={}, @category_word_count={}, @language="es", @auto_categorize=false, @enable_threshold=false, @threshold=0.0>
> classifier.train_bueno "pastel ron gatos"
=> [
    [0] "pastel ron gatos"
]
> classifier.train_malo "cosas malas como lluvia, dolor, enfermidad"
=> [
    [0] "cosas malas como lluvia, dolor, enfermidad"
]
> classifier.classify "lluvia"
=> "Malo"