Merge pull request #10 from bkimminich/training-format

Update training data format
juice-shop · Sep 6, 2020 · 4b3800f · 4b3800f
2 parents 50bc432 + 1568993
commit 4b3800f
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 52 deletions.
diff --git a/factory.js b/factory.js
@@ -15,11 +15,13 @@ var users = {
 }
 
 function train () {
-  trainingSet.intents.map((query) => {
-    model.addDocument(trainingSet.lang, query.question, query.intent)
-  })
-  trainingSet.answers.map((query) => {
-    model.addAnswer(trainingSet.lang, query.intent, query.answer)
+  trainingSet.data.map((query) => {
+    query.utterances.map((utterance) => {
+      model.addDocument(trainingSet.lang, utterance, query.intent)
+    })
+    query.answers.map((answer) => {
+      model.addAnswer(trainingSet.lang, query.intent, answer)
+    })
   })
   return model.train().then(() => { training.state = true })
 }

diff --git a/test/initialize-spec.js b/test/initialize-spec.js
@@ -4,58 +4,33 @@ const juice = require('../index')
 
 const trainingSet = {
   lang: 'en',
-  intents: [
-    {
-      question: 'goodbye for now',
-      intent: 'greetings.bye'
-    },
-    {
-      question: 'bye bye take care',
-      intent: 'greetings.bye'
-    },
-    {
-      question: 'hello',
-      intent: 'greetings.hello'
-    },
-    {
-      question: 'hi',
-      intent: 'greetings.hello'
-    },
-    {
-      question: 'howdy',
-      intent: 'greetings.hello'
-    },
-    {
-      question: 'how much is X',
-      intent: 'queries.productprice'
-    },
-    {
-      question: 'how much does X cost',
-      intent: 'queries.productprice'
-    }
-  ],
-  answers: [
+  data: [
     {
       intent: 'greetings.bye',
-      answer: {
-        action: 'response',
-        body: 'Ok Cya'
-      }
+      utterances: [
+        'goodbye for now',
+        'bye bye take care'
+      ],
+      answers: [
+        {
+          action: 'response',
+          body: 'Ok Cya'
+        }
+      ]
     },
     {
       intent: 'greetings.hello',
-      answer: {
-        action: 'response',
-        body: 'Hello <customer-name>'
-      }
-    },
-    {
-      intent: 'queries.productprice',
-      answer: {
-        action: 'function',
-        handler: 'productPrice',
-        body: ''
-      }
+      utterances: [
+        'hello',
+        'hi',
+        'howdy'
+      ],
+      answers: [
+        {
+          action: 'response',
+          body: 'Hello <customer-name>'
+        }
+      ]
     }
   ]
 }