ไลบรารี JavaScript ที่นำการค้นหาเวกเตอร์และผ้าขี้ริ้วไปยังเบราว์เซอร์ของคุณ!
| - ผู้ตรวจสอบกระดาษ ML | - เพิ่มประสิทธิภาพ | - ผู้ช่วย AI รับผิดชอบ |
Mememo เป็นไลบรารี JavaScript ที่ปรับใช้เทคนิคการค้นหาเพื่อนบ้านที่ใกล้ที่สุดโดยประมาณ HNSW กับสภาพแวดล้อมของเบราว์เซอร์ พัฒนาด้วยเทคโนโลยีเว็บที่ทันสมัยและพื้นเมืองเช่น IndexedDB และ Web Workers ชุดเครื่องมือของเราใช้ประโยชน์จากความสามารถด้านฮาร์ดแวร์ฝั่งไคลเอ็นต์เพื่อให้นักวิจัยและนักพัฒนาสามารถค้นหาได้อย่างมีประสิทธิภาพผ่านเวกเตอร์มิติสูงหลายล้านตัวในเบราว์เซอร์ Mememo ช่วยให้โอกาสการออกแบบและการวิจัยใหม่ที่น่าตื่นเต้นเช่นการสร้างเนื้อหาส่วนตัวและส่วนตัวและการสร้างต้นแบบแบบโต้ตอบดังที่แสดงในตัวอย่างแอปพลิเคชัน RAG Playground ของเรา
Mememo รองรับทั้งสภาพแวดล้อมของเบราว์เซอร์และ node.js ในการติดตั้ง Mememo คุณสามารถใช้ npm :
npm install mememoจากนั้นคุณสามารถสร้างดัชนีเวกเตอร์และทำการค้นหาเพื่อนบ้านที่ใกล้ที่สุดโดยประมาณสองฟังก์ชั่น:
// Import the HNSW class from the MeMemo module
import { HNSW } from 'mememo' ;
// Creating a new index
const index = new HNSW ( { distanceFunction : 'cosine' } ) ;
// Inserting elements into our index in batches
let keys : string [ ] ;
let values : number [ ] [ ] ;
await index . bulkInsert ( keys , values ) ;
// Find k-nearest neighbors
let query : number [ ] ;
const { keys , distances } = await index . query ( query , k ) ; โคลนหรือดาวน์โหลดที่เก็บนี้:
git clone [email protected]:poloclub/mememo.gitติดตั้งการอ้างอิง:
npm installใช้ VATEST สำหรับการทดสอบหน่วย:
npm run test
โคลนหรือดาวน์โหลดที่เก็บนี้:
git clone [email protected]:poloclub/mememo.gitนำทางไปยังโฟลเดอร์ตัวอย่าง:
cd ./examples/rag-playgroundติดตั้งการอ้างอิง:
npm installจากนั้นเรียกใช้คำอธิบายสินเชื่อ:
npm run dev
นำทางไปยัง LocalHost: 3000 คุณควรเห็นผู้อธิบายสามคนทำงานในเบราว์เซอร์ของคุณ :)
Mememo ถูกสร้างขึ้นโดย Jay Wang และ Polo Chau
หากต้องการเรียนรู้เพิ่มเติมเกี่ยวกับ Mememo ลองดูรายงานการวิจัยที่เผยแพร่ที่ SIGIR'24
@inproceedings { wangMeMemoOndeviceRetrieval2024 ,
title = { {{MeMemo}}: {{On-device Retrieval Augmentation}} for {{Private}} and {{Personalized Text Generation}} } ,
booktitle = { Proceedings of the 47th {{International ACM SIGIR Conference}} on {{Research}} and {{Development}} in {{Information Retrieval}} } ,
author = { Wang, Zijie J. and Chau, Duen Horng } ,
year = { 2024 } ,
urldate = { 2024-06-26 } ,
langid = { english }
}ซอฟต์แวร์นี้มีอยู่ภายใต้ใบอนุญาต MIT
หากคุณมีคำถามใด ๆ อย่าลังเลที่จะเปิดปัญหาหรือติดต่อ Jay Wang