Recommendation engine based on collaborative filtering
This API is a lightweight implementation of collaborative filtering for Node.js. The only dependency is Math.js. Currently it supports generating recommendations with Jaccard similarity.
npm i collaborative-filter
In your project, simply include the module.
const recommend = require('collaborative-filter')
To run the provided example.
node examples/demo.js
The input for the engine is an array matrix which defines the ratings of the users in the database. It should be a matrix containing of 0:s (not rated) and 1:s (liked the item) and follow this format.
I0 I1 I2 . . .[U0 [1 1 1 . . .],U1 [1 0 1 . . .],U2 [1 0 0 . . .],. [. . . . . .],. [. . . . . .],. [. . . . . .],]
In javascript, this could look something like this
const ratings = [[1, 1, 1],[1, 0, 1],[1, 0, 0],];
If you want to run the whole collaborative filter, you would do this:
const recommend = require('../lib/cf_api.js');const result = recommend.cFilter(ratings, 2);
where 2 is the user index. The output of this with ratings matrix as above, would be an array [2, 1]. This tells us that item 2 is the most appropriate recommendation followed by item 1.
You could also run the filtering process by calling the global API functions individually.
const recommend = require('collaborative-filter');const coMatrix = recommend.createCoMatrix(ratings, numUsers, numItems);const result = recommend.getRecommendations(ratings, coMatrix, userIndex);
which results in the same array as before. This could be useful when you do not need to generate the co-occurrence matrix multiple times. For instance, if you want recommendations for multiple users, you do not need to generate the co-occurrence matrix multiple times, saving you processing time.
The Cold start problem) is something that can make or break a recommendation application. If you don’t have enough data, it’s hard to draw any conclusions, especially if the number of items is large.
In the file cf_api.js in the lib directory, there is a flag ONLY_RECOMMEND_FROM_SIMILAR_TASTE. If you put this to 0, you will get recommendations from users which don’t necessarily have similar taste as you (these will however be lower ranked than recommendations from people with similar taste). This option is available if you consider a cold start something that will make your service seem poor. With this flag enabled, you will never receive a recommendation from someone who has no similarity with you.
You can also disable the NORMALIZE_ON_POPULARITY flag, which in turn ensures that the co-occurrence matrix is not normalized.
Submit a pull request if you want to contribute. We follow the Airbnb JavaScript Style Guide.
MIT
