(vectors, query, k=5)
| 297 | |
| 298 | |
| 299 | def bruteforce_search(vectors, query, k=5): |
| 300 | similarities = [] |
| 301 | for vector in vectors: |
| 302 | similarity = cosine_similarity(query["values"], vector["values"]) |
| 303 | similarities.append((vector["id"], similarity)) |
| 304 | |
| 305 | similarities.sort(key=lambda x: x[1], reverse=True) |
| 306 | |
| 307 | return similarities[:k] |
| 308 | |
| 309 | |
| 310 | def generate_vectors( |
no test coverage detected