search

jekyll / classifier-reborn

A general classifier module to allow Bayesian and other types of classifications. A fork of cardmagic/classifier.
https://jekyll.github.io/classifier-reborn/
GNU Lesser General Public License v2.1
548 stars 109 forks source link

Multiple clasifications #165

Open gugaiz opened 6 years ago

gugaiz commented 6 years ago

Hi I tried to do a test with multiple classifications (i.e. classifier = ClassifierReborn::Bayes.new 'Social', 'Email', 'Direct', 'Organic','Other') but it does not seems to work. I mean, even after training it always keep returning "Social" for different sentences. Should it work ok? Am I doing something wrong? or is this not supported yet?

Ch4s3 commented 6 years ago

Which version are you using? The readme is focused on master, which I haven't released just yet.

gugaiz commented 6 years ago

I made it work for me. I forked and changed the way "Infinity" is being used, just in case you want to have a look to the change.

Ch4s3 commented 6 years ago

@gugaiz could you link to what you're doing or open a pr here?

Ch4s3 commented 6 years ago

Ahh, I just looked at the fork https://github.com/gugaiz/classifier-reborn/commit/5fdd6b89bf08d6657cdb822dd479d86252661068 and I see what you're doing @gugaiz. This may be a bug WRT thresholds. I'd be happy to look at a pr.