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

Improved recall test

Browse Source
This commit is contained in:
Andrew Kane
2022-02-13 03:08:39 -08:00
parent 310809d0e5
commit 9549d93260
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
test/t/003_recall.pl
View File

@@ -7,6 +7,7 @@ use Test::More tests => 3;
my $node; my $node;
my @queries = (); my @queries = ();
my @expected = (); my @expected = ();
my $limit = 30;
sub test_recall sub test_recall
{ {
@@ -18,7 +19,7 @@ sub test_recall
my $actual = $node->safe_psql("postgres", qq( my $actual = $node->safe_psql("postgres", qq(
SET enable_seqscan = off; SET enable_seqscan = off;
SET ivfflat.probes = $probes; SET ivfflat.probes = $probes;
SELECT i FROM tst ORDER BY v <-> '$queries[$i]' LIMIT 10; SELECT i FROM tst ORDER BY v <-> '$queries[$i]' LIMIT $limit;
)); ));
my @actual_ids = split("\n", $actual); my @actual_ids = split("\n", $actual);
my %actual_set = map { $_ => 1 } @actual_ids; my %actual_set = map { $_ => 1 } @actual_ids;
@@ -58,7 +59,7 @@ for (1..20) {
# Get exact results # Get exact results
foreach (@queries) { foreach (@queries) {
my $res = $node->safe_psql("postgres", "SELECT i FROM tst ORDER BY v <-> '$_' LIMIT 10;"); my $res = $node->safe_psql("postgres", "SELECT i FROM tst ORDER BY v <-> '$_' LIMIT $limit;");
push(@expected, $res); push(@expected, $res);
} }