在 Swift 中使用 Faiss。
基于 Faiss Mobile 和 OpenMP Mobile。
命令行演示
$ swift run swift-faiss <subcommand> <options>
可用的子命令
flat: 创建一个FlatIndex，向其添加向量并搜索最相似的句子。ivfflat: 创建一个IVFFlatIndex，训练并向其添加向量并搜索最相似的句子。pq: 创建一个PQIndex，训练并向其添加向量并搜索最相似的句子。clustering: k-means 聚类示例。
命令行帮助
$ swift run swift-faiss --help
在你自己的代码中
import SwiftFaiss
let embeddings: [[Float]] = [
[0.1, 0.2, 0.3],
[0.4, 0.5, 0.6],
[0.7, 0.8, 0.9],
[1.0, 1.1, 1.2],
[1.3, 1.4, 1.5],
[1.6, 1.7, 1.8]
]
let d = embeddings[0].count
let index = try FlatIndex(d: d, metricType: .l2)
try index.add(embeddings)
let result = try index.search([[0.1, 0.5, 0.9]], k: 2)
// do something with result
https://github.com/jkrukowski/UseSwiftFaiss 包含一个 iOS 示例。
你可以使用 Swift Package Manager 并在 Package.swift 中指定依赖，通过添加
.package(url: "https://github.com/jkrukowski/SwiftFaiss.git", from: "0.0.7")
$ swift package plugin --allow-writing-to-package-directory swiftformat
$ swift test