Skip to content

Instantly share code, notes, and snippets.

View cutecutecat's full-sized avatar

cutecutecat cutecutecat

  • Southern University of Science and Technology
  • Shanghai, China
View GitHub Profile
@cutecutecat
cutecutecat / filter.py
Created May 30, 2025 12:04
filter draw plot script
from matplotlib import ticker
import numpy as np
import pandas as pd
import seaborn as sns
import matplotlib.pyplot as plt
QPS = [13.42, 12.32, 11.23, 8.75, 27.65, 25.36, 20.41, 16.66]
original_data = {
"acc": [
@cutecutecat
cutecutecat / dump.py
Last active January 23, 2025 07:12
kmeans toolkit
import argparse
import datetime
import logging
import math
import psutil
import numpy as np
import psycopg
from tqdm import tqdm
from pgvector.psycopg import register_vector
@cutecutecat
cutecutecat / draw_benchmark.py
Last active November 26, 2024 02:51
benchmark graph
import seaborn as sns
import matplotlib.pyplot as plt
qps = [25.69, 18.91, 17.13, 16.18, 14.45, 7.24, 4.51]
recall = [0.9079, 0.9461, 0.9497, 0.9535, 0.9608, 0.9650, 0.9672]
p99_latency = [148.25, 260.68, 269.5, 302.29, 456.09, 1042.41, 1675.42]
fig, (ax1, ax3) = plt.subplots(1, 2, figsize=(12, 5), gridspec_kw={'width_ratios': [20, 7]})
@cutecutecat
cutecutecat / bench.py
Last active January 26, 2025 21:33
rabbithole benchmark on external centroids
from os.path import join
import os
import time
import argparse
from pathlib import Path
from tqdm import tqdm
import psycopg
import h5py
from pgvecto_rs.psycopg import register_vector
@cutecutecat
cutecutecat / citus_pgvectors.py
Last active September 5, 2024 14:07
Citus with pgvecto.rs
import numpy as np
from pgvecto_rs.psycopg import register_vector
import psycopg
# generate random data
rows = 100000
dimensions = 128
embeddings = np.random.rand(rows, dimensions)
categories = np.random.randint(100, size=rows).tolist()
queries = np.random.rand(10, dimensions)