EnergeticAI 是一款专为 Node.js 应用设计的开源人工智能工具。它基于 TensorFlow.js 进行了优化,适用于无服务器环境,具有快速冷启动、小模块尺寸和预训练模型等优点。使用 EnergeticAI,您可以在几秒钟内完成安装,并通过对商业友好的许可进行扩展。
该工具的预训练嵌入可用于推荐等多种应用。其模型和库能让您快速上手,利用预训练模型迅速开展工作。通过句子嵌入,您可以构建推荐系统等功能。文档中还提供了分类器的相关内容,您可以使用少量训练示例将文本分类。此外,虽然语义搜索功能目前仍在计划中,但它将通过基于意义的问答模型为您提供答案。
在性能方面,EnergeticAI 在最大程度提高冷启动性能的同时,将模块大小最小化。与 TensorFlow.js 相比,它的速度快了高达 67 倍。在无服务器函数中,推理速度主要由冷启动速度决定。此基准测试在 M1 Max Macbook Pro 上初始化模型并为一个 5 句段落计算嵌入。
EnergeticAI 的可用性也很高,您可以在几秒钟内从 NPM 下载并安装:npm install @energetic-ai/core
,但需要 Node 18+。该工具采用 Apache 2.0 许可,不过其依赖项可能有所不同。