ai-zero-shot-classifier/dist/utils/similarityFunctions/index.js

🧠 powerful JavaScript library that leverages advanced AI embeddings to perform zero-shot text classification. Whether you're dealing with unlabelled data or seeking to classify text against dynamic and user-defined labels, this library provides a seamles

"use strict";

Object.defineProperty(exports, "__esModule", {
  value: true
});
exports.getSimilarityFunction = exports.default = void 0;
var _cosineSimilarity = _interopRequireDefault(require("./cosineSimilarity"));
var _dotProduct = _interopRequireDefault(require("./dotProduct"));
var _euclideanDistance = _interopRequireDefault(require("./euclideanDistance"));
function _interopRequireDefault(e) { return e && e.__esModule ? e : { default: e }; }
/**
 * Similarity function mapping
 */
var similarityFunctions = {
  cosine: _cosineSimilarity.default,
  dot: _dotProduct.default,
  euclidean: _euclideanDistance.default
};
/**
 * Get similarity function based on the similarity type
 *
 * @param similarity
 * @returns Function
 */
var getSimilarityFunction = exports.getSimilarityFunction = function getSimilarityFunction(similarity) {
  var fn = similarityFunctions[similarity];
  if (!fn) {
    throw new Error("Invalid similarity function: \"".concat(similarity, "\""));
  }
  return fn;
};
var _default = exports.default = similarityFunctions;
//# sourceMappingURL=index.js.map