Assign likelihoods based on 'robust model', tied to user/title/body.

Author: kildrens

src/resolvers/index.ts

@@ -1,6 +1,6 @@
 import * as scuid from 'scuid';
 
-import { generateAuthToken, getUserId, probabilityHistogram, topicWords } from '../utils';
+import { generateAuthToken, getUserId, likelyTopics } from '../utils';
 
 const DEFAULT_COUNT = 25;
 
@@ -66,42 +66,57 @@ export default {
     },
 
     // refactor user relation into Class
-    Post: (parent, { id }, { faker }) => ({
-      id,
-      title: faker.random.words(),
-      body: faker.lorem.paragraphs(),
-      published: faker.random.boolean(),
-      createdAt: faker.date.past(),
-      author: {
-        id: scuid(),
-        firstName: faker.name.firstName(),
-        lastName: faker.name.lastName(),
-        email: faker.internet.email(),
-        avatar: faker.image.avatar()
-      }
-    }),
-
-    // refactor user relation into Class
-    allPosts: (parent, { count }, { faker }) => {
-      return new Array(count).fill(0).map(_ => ({
-        id: scuid(),
-        title: faker.random.words(),
-        body: faker.lorem.sentences(),
+    Post: (parent, { id }, { faker }) => {
+      const title = faker.random.words();
+      const body = faker.lorem.paragraphs();
+      const firstName = faker.name.firstName();
+      const lastName = faker.name.lastName();
+      return {
+        id,
+        title,
+        body,
         published: faker.random.boolean(),
         createdAt: faker.date.past(),
         author: {
           id: scuid(),
-          firstName: faker.name.firstName(),
-          lastName: faker.name.lastName(),
+          firstName,
+          lastName,
           email: faker.internet.email(),
           avatar: faker.image.avatar()
         },
-        likelyTopics: probabilityHistogram(10)
-          .map((likelihood, i) => ({
-            label: topicWords[i],
-            likelihood: likelihood
-          }))
-      }));
+        likelyTopics: likelyTopics(
+          `${firstName} ${lastName}`,
+          title, body
+        )
+      };
+    },
+
+    // refactor user relation into Class
+    allPosts: (parent, { count }, { faker }) => {
+      return new Array(count).fill(0).map(_ => {
+        const title = faker.random.words();
+        const body = faker.lorem.paragraphs();
+        const firstName = faker.name.firstName();
+        const lastName = faker.name.lastName();
+        return {
+          id: scuid(),
+          title,
+          body,
+          published: faker.random.boolean(),
+          createdAt: faker.date.past(),
+          author: {
+            id: scuid(),
+            firstName,
+            lastName,
+            email: faker.internet.email(),
+            avatar: faker.image.avatar()
+          },
+          likelyTopics: likelyTopics(
+            `${firstName} ${lastName}`,
+            title, body
+          )
+        }
+      });
     }
   },
 

src/utils.ts

@@ -17,28 +17,48 @@ export const getUserId = ctx => {
   throw new Error('Not Authorized');
 };
 
-// Generate a uniform probability histogram
-export const probabilityHistogram = bins => {
-  const independent = []
-  let total = 0
-  for (let i = 0; i < bins; i++) {
-    const weight = Math.random()
-    total += weight
-    independent.push(weight)
-  }
-  return independent.map(weight => weight / total)
-}
-
 // Ten randomly-chosen but consistent topic nouns.
 export const topicWords = [
-  'drama',
   'wedding',
-  'management',
-  'security',
-  'quality',
+  'sport',
   'fishing',
-  'client',
+  'shopping',
+  'security',
+  'management',
+  'community',
+  'celebrity',
   'birthday',
-  'disaster',
   'potato'
-]
+];
+
+export const likelyTopics = (user, title, body) => {
+  // create a custom order over the topics, based on letters as they appear in
+  // the user's name, post title, and post body, for variability
+  const alphabet = `${title} ${user} ${body}`;
+  const topicLabels = topicWords.sort((a, b) => {
+    for (let i = 0; i < Math.min(a.length, b.length); i++) {
+      const aValue = alphabet.indexOf(a[i]);
+      const bValue = alphabet.indexOf(b[i]);
+      if (aValue !== bValue) {
+        return bValue - aValue;
+      }
+    }
+    return b.length - a.length;
+  })
+  // compute a biased probability distribution
+  const likelihoods = [];
+  let total = 0;
+  for (let i = 0; i < topicLabels.length; i++) {
+    const weight = Math.random() * Math.log(i + 2);
+    total += weight;
+    likelihoods.push(weight);
+  }
+  likelihoods.reverse();
+  // and associate each likelihood with a probability
+  return likelihoods.map((weight, i) => {
+    return {
+      likelihood: weight / total,
+      label: topicLabels[i]
+    }
+  }).sort((a, b) => b.likelihood - a.likelihood);
+};