feat: closes #2 accept string as category

Author: NathanFirmo

src/classifier.ts

@@ -22,15 +22,24 @@ export class Classifier {
     this.options = options
   }
 
-  learn(sentence: string, classifications: string[]) {
+  learn(sentence: string, inputs: string | string[]) {
+    const classifications = Array.isArray(inputs)
+      ? inputs
+      : [inputs]
+
     classifications.forEach((classification) => {
       let categorie = this.categories.find(
         (categorie) => categorie.name === classification
       )
-      const relatedCategories = classifications.filter(item => item !== classification)
+      const relatedCategories = classifications.filter(
+        (item) => item !== classification
+      )
       if (categorie) categorie.addSentence(sentence, relatedCategories)
       else {
-        categorie = new Category(classification).addSentence(sentence, relatedCategories)
+        categorie = new Category(classification).addSentence(
+          sentence,
+          relatedCategories
+        )
         this.categories.push(categorie)
       }
     })
@@ -65,7 +74,9 @@ export class Classifier {
       Object.values(classification).reduce(sumFunc, 0) + unknownScore
     result.unknown = this.options?.percentualReturn
       ? toPercent(!!relevancySum ? unknownScore / relevancySum : unknownScore)
-      : !!relevancySum ? unknownScore / relevancySum : unknownScore
+      : !!relevancySum
+      ? unknownScore / relevancySum
+      : unknownScore
 
     for (const [name, relevancy] of Object.entries(classification)) {
       const value = relevancySum ? relevancy / relevancySum : 0

tests/classifier.spec.ts

@@ -108,5 +108,23 @@ describe('Classifier', () => {
         })
       )
     })
+
+    it('should be able to accept string as category to classify sentences', () => {
+      classifier.resetKnowledge()
+      classifier.learn('I like cats', 'animal')
+      classifier.learn('Cats are cool', 'animal')
+      classifier.learn('Dogs are noisy', 'animal')
+      classifier.learn('I love animals', 'animal')
+      classifier.learn('I like my horse', 'animal')
+      classifier.learn('Chocolate is good', 'food')
+      classifier.learn('I eat apple', 'food')
+      classifier.learn('Juice is very good', 'food')
+      classifier.learn('Brazilians eat rice and beans', 'food')
+      classifier.learn('Bananas are good for health', 'food')
+
+      expect(classifier.classify('Apple juice is awesome')).toEqual(
+        expect.objectContaining({ unknown: 0.2, animal: 0, food: 0.8 })
+      )
+    })
   })
 })