Function main

sample/python/symqg_indexing.py:16–33

(args=None)

QUANTIZATION_BITS = 0  # 0 for vanilla QG, or 4/8 for QG-quant
NUM_THREADS = 16  # number of threads for build
INIT = "pipnn"  # "pipnn" or "random" initialization of SymphonyQG
# ──────────────────────────────────────────────


def main(args=None) -> None:
    # 1. Load data
    data = read_fvecs(args.data_file)
    print(f"Data shape: {data.shape}")

    # 3. Build SymphonyQG index
    n, dim = data.shape
    print(
        f"\nBuilding SymphonyQG index: n={n}, dim={dim}, MaxDegree={args.max_degree}, "
        f"ef={args.ef_construction}, metric={args.metric}, "
        f"quantization_bits={args.quantization_bits}, init={args.init}"
    )

    idx = SymqgIndex(
        dim=dim,
        max_degree=args.max_degree,
        metric=args.metric,