zlei6/ragflow
1
0
Fork 0
mirror of https://github.com/infiniflow/ragflow.git synced 2026-07-25 01:43:27 +08:00
Code Issues Packages Projects Releases Wiki Activity

refa: improve tree clustering (#17285)

Browse Source
This commit is contained in:
buua436
2026-07-23 17:49:13 +08:00
committed by GitHub
parent 7b64e4dc5d
commit d4a8c91f3c
6 changed files with 195 additions and 74 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
test/unit_test/rag/test_raptor_psi_tree_builder.py
View File

@@ -18,8 +18,6 @@ import importlib
import os
import sys
import types
from unittest.mock import MagicMock
import pytest
np = pytest.importorskip("numpy")
@@ -54,11 +52,12 @@ def raptor_module(monkeypatch):
return np.ones((len(embeddings), 1))
class DummyAgglomerativeClustering:
def __init__(self, n_clusters=None, distance_threshold=None, compute_distances=False, linkage="ward"):
def __init__(self, n_clusters=None, distance_threshold=None, compute_distances=False, linkage="ward", metric="euclidean"):
self.n_clusters = n_clusters
self.distance_threshold = distance_threshold
self.compute_distances = compute_distances
self.linkage = linkage
self.metric = metric
self.distances_ = np.array([0.1, 0.2, 1.0])
def fit(self, embeddings):