retrieval/src/Bm25/vec_space/vec_space.js

Full text search engine in js. Features BM25 ranking function that can be tuned.

/*
 * Convert raw data to vector space attributes.
 */
const tfMaps = require("./lib/tf.js");
const relDocLengths = require("./lib/doc_len.js");

module.exports = (function() {
  return function(corpusMatr) {
				return {
							docLens: relDocLengths(corpusMatr),
						  	docs: tfMaps(corpusMatr)
						 };
			};
})();