detector_benchmark.create_dataset

Functions

choose_dataset(dataset_name, dataset_size, ...)
choose_generator(model_name, gen_params, device)
choose_watermarking_scheme(cfg, ...)
choose_attack(cfg, attack_type, gen_model, ...[, ...])
create_dataset(cfg)
main()

Module Contents

detector_benchmark.create_dataset.choose_dataset(dataset_name: str, dataset_size: int, max_sample_len: int, prefix_size: int)

detector_benchmark.create_dataset.choose_generator(model_name: str, gen_params, device: str)

detector_benchmark.create_dataset.choose_watermarking_scheme(cfg: omegaconf.DictConfig, watermarking_scheme_name: str, gen, model_config)

detector_benchmark.create_dataset.choose_attack(cfg: omegaconf.DictConfig, attack_type: str, gen_model, model_config, max_sample_len, watermarking_scheme=None, paraphraser_model=None, paraphraser_config=None)

detector_benchmark.create_dataset.create_dataset(cfg: omegaconf.DictConfig)

detector_benchmark.create_dataset.main()