Repository navigation
Expand file tree
/
Copy pathsearch.ts
More file actions
103 lines (90 loc) · 2.24 KB
/
Copy pathsearch.ts
File metadata and controls
103 lines (90 loc) · 2.24 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
export type GalleryEmbedding = {
id: string;
label: string;
vector: number[];
imagePath?: string;
floor?: string;
zone?: string;
x?: number;
y?: number;
};
export type SearchResult = GalleryEmbedding & {
score: number;
rank: number;
};
export type SearchConfidenceLevel = "confident" | "ambiguous" | "uncertain";
export type SearchConfidence = {
level: SearchConfidenceLevel;
topScore: number;
margin: number;
reason: string;
};
type ScoredResult = {
score: number;
[key: string]: unknown;
};
function dot(a: number[], b: number[]): number {
let value = 0;
for (let index = 0; index < a.length; index += 1) {
value += a[index] * b[index];
}
return value;
}
function norm(vector: number[]): number {
return Math.sqrt(dot(vector, vector));
}
export function cosineSimilarity(a: number[], b: number[]): number {
if (a.length !== b.length) {
throw new Error(`벡터 차원이 일치하지 않습니다: query=${a.length}, candidate=${b.length}`);
}
const denominator = norm(a) * norm(b);
if (denominator === 0) {
return 0;
}
return dot(a, b) / denominator;
}
export function searchEmbeddings(
query: number[],
gallery: GalleryEmbedding[],
topK: number,
): SearchResult[] {
return gallery
.map((candidate) => ({
...candidate,
score: cosineSimilarity(query, candidate.vector),
rank: 0,
}))
.sort((left, right) => right.score - left.score)
.slice(0, topK)
.map((result, index) => ({
...result,
rank: index + 1,
}));
}
export function classifySearchConfidence(results: ScoredResult[]): SearchConfidence {
const topScore = results[0]?.score ?? 0;
const secondScore = results[1]?.score ?? 0;
const margin = Number((topScore - secondScore).toFixed(6));
if (topScore < 0.72) {
return {
level: "uncertain",
topScore,
margin,
reason: "top-1 점수가 낮아 위치를 확정하지 않습니다.",
};
}
if (margin < 0.05) {
return {
level: "ambiguous",
topScore,
margin,
reason: "top-1과 top-2 후보가 가까워 다중 후보로 봅니다.",
};
}
return {
level: "confident",
topScore,
margin,
reason: "top-1 점수와 margin이 기준을 통과했습니다.",
};
}