- 博客(5)
- 收藏
- 关注
原创 Faiss库与KNN算法
Faiss是由Facebook AI Research团队开发的一个库,旨在高效地进行大规模向量相似性搜索。它支持CPU,且可以GPU进行加速(需要下载GPU版本),非常适合处理大量高维数据。Faiss提供了多种索引类型,以适应不同的需求,从简单的平面索引(Flat Index)到更复杂的倒排文件索引(IVF)和乘积量化索引(PQ)
2025-09-06 11:16:14
373
原创 K-近邻算法(KNN算法)——附带pytorch框架下的代码
交叉验证是一种对于模型的评估手段,通过反复训练模型以确保模型的可靠交叉验证时会将数据分为多份,每份称为一折。
2025-05-20 23:51:55
1488
原创 KD树——附代码实现
kd树是一种树形结构,树的每个节点存放一个k维数据,某一节点的子节点可以看作是由过该节点一个平面切割后产生的,其具体演示如图:(找到中间点,比中间点小的放到一起,比中间点大的放到一起)节点信息包括:(1)节点所处点点值:比较距离时候要用(2)节点维度:方便后续比较距离(3)节点左枝节点(4)节点右枝节点#退出递归条件#求维度#循环维度#求出中位数所在下标points:需要构建KD树的数据集depth:维度,默认值为0设置推出递归的条件,循环到空节点。
2025-05-18 22:07:43
1673
1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人
RSS订阅