mumble-cleanup / scripts /01_download.py
adikuma's picture
initial upload: cleanup code and 688-pair seed dataset
fd0b01f verified
Raw
History Blame Contribute Delete
729 Bytes
# load the hand-crafted seed pairs at data/seed/synthetic_pairs.jsonl and
# split them stratified-by-category into train/val/test under data/pairs/.
import argparse
from pathlib import Path
from cleanup.config import load_data_config
from cleanup.data.download import build_and_save
def main() -> None:
parser = argparse.ArgumentParser()
parser.add_argument("--config", default="configs/data.yaml")
parser.add_argument("--out", default="data/pairs")
parser.add_argument("--smoke", action="store_true", help="tiny subset for wiring tests")
args = parser.parse_args()
cfg = load_data_config(args.config)
build_and_save(cfg, Path(args.out), smoke=args.smoke)
if __name__ == "__main__":
main()