liuyu/pgvector
1
0
Fork 0
mirror of https://github.com/pgvector/pgvector.git synced 2026-07-06 12:40:56 +08:00
Code Issues Packages Projects Releases Wiki Activity

Improved recall test [skip ci]

Browse Source
This commit is contained in:
Andrew Kane
2024-03-31 22:38:17 -07:00
parent caca8990fc
commit 3fc734fb36
1 changed files with 11 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
test/t/020_hnsw_halfvec_build_recall.pl
View File

@@ -8,6 +8,8 @@ my $node;
my @queries = (); my @queries = ();
my @expected; my @expected;
my $limit = 20; my $limit = 20;
my $dim = 10;
my $array_sql = join(",", ('random()') x $dim);
sub test_recall sub test_recall
{ {
@@ -52,18 +54,20 @@ $node->start;
# Create table # Create table
$node->safe_psql("postgres", "CREATE EXTENSION vector;"); $node->safe_psql("postgres", "CREATE EXTENSION vector;");
$node->safe_psql("postgres", "CREATE TABLE tst (i int4, v halfvec(3));"); $node->safe_psql("postgres", "CREATE TABLE tst (i int4, v halfvec($dim));");
$node->safe_psql("postgres", $node->safe_psql("postgres",
"INSERT INTO tst SELECT i, ARRAY[random(), random(), random()] FROM generate_series(1, 10000) i;" "INSERT INTO tst SELECT i, ARRAY[$array_sql] FROM generate_series(1, 10000) i;"
); );
# Generate queries # Generate queries
for (1 .. 20) for (1 .. 20)
{ {
my $r1 = rand(); my @r = ();
my $r2 = rand(); for (1 .. $dim)
my $r3 = rand(); {
push(@queries, "[$r1,$r2,$r3]"); push(@r, rand());
}
push(@queries, "[" . join(",", @r) . "]");
} }
# Check each index type # Check each index type
@@ -90,7 +94,7 @@ for my $i (0 .. $#operators)
)); ));
# Test approximate results # Test approximate results
my $min = $operator eq "<#>" ? 0.80 : ($operator eq "<=>" ? 0.80 : 0.99); my $min = $operator eq "<#>" ? 0.80 : 0.99;
test_recall($min, $operator); test_recall($min, $operator);
$node->safe_psql("postgres", "DROP INDEX idx;"); $node->safe_psql("postgres", "DROP INDEX idx;");