pe

• pe package
  • Subpackages
    • pe.api package
      • API
        • API.random_api()
        • API.variation_api()
      • Avatar
        • Avatar.__init__()
        • Avatar._get_params_from_avatar()
        • Avatar._get_random_image()
        • Avatar._get_variation_image()
        • Avatar._svg_to_numpy()
        • Avatar.random_api()
        • Avatar.variation_api()
      • DrawText
        • DrawText.__init__()
        • DrawText._create_image()
        • DrawText._get_random_image()
        • DrawText._get_variation_image()
        • DrawText.random_api()
        • DrawText.variation_api()
      • ImprovedDiffusion
        • ImprovedDiffusion.__init__()
        • ImprovedDiffusion.random_api()
        • ImprovedDiffusion.variation_api()
      • ImprovedDiffusion270M
        • ImprovedDiffusion270M.CHECKPOINT_URL
        • ImprovedDiffusion270M.__init__()
      • LLMAugPE
        • LLMAugPE.__init__()
        • LLMAugPE._blank_sample()
        • LLMAugPE._construct_prompt()
        • LLMAugPE.random_api()
        • LLMAugPE.variation_api()
      • NearestImage
        • NearestImage.__init__()
        • NearestImage._build_nearest_neighbor_graph()
        • NearestImage.random_api()
        • NearestImage.variation_api()
      • StableDiffusion
        • StableDiffusion.__init__()
        • StableDiffusion.random_api()
        • StableDiffusion.variation_api()
      • Subpackages
        • pe.api.image package
          • Subpackages
            • pe.api.image.improved_diffusion_lib package
          • Submodules
            • pe.api.image.avatar_api module
            • pe.api.image.draw_text_api module
            • pe.api.image.improved_diffusion_api module
            • pe.api.image.nearest_image_api module
            • pe.api.image.stable_diffusion_api module
        • pe.api.text package
          • Submodules
            • pe.api.text.llm_augpe_api module
      • Submodules
        • pe.api.api module
          • API
            • API.random_api()
            • API.variation_api()
        • pe.api.util module
          • ConstantList
    • pe.callback package
      • Callback
        • Callback.__call__()
      • ComputeFID
        • ComputeFID.__call__()
        • ComputeFID.__init__()
      • ComputePrecisionRecall
        • ComputePrecisionRecall.__call__()
        • ComputePrecisionRecall.__init__()
      • DPImageBenchClassifyImages
        • DPImageBenchClassifyImages.__call__()
        • DPImageBenchClassifyImages.__init__()
        • DPImageBenchClassifyImages._get_data_loader()
        • DPImageBenchClassifyImages._get_images_and_label_from_data()
        • DPImageBenchClassifyImages._get_model()
        • DPImageBenchClassifyImages.evaluate()
      • SampleImages
        • SampleImages.__call__()
        • SampleImages.__init__()
      • SaveAllImages
        • SaveAllImages.__call__()
        • SaveAllImages.__init__()
        • SaveAllImages._save_image()
      • SaveCheckpoints
        • SaveCheckpoints.__call__()
        • SaveCheckpoints.__init__()
        • SaveCheckpoints._get_checkpoint_path()
      • SaveTextToCSV
        • SaveTextToCSV.__call__()
        • SaveTextToCSV.__init__()
        • SaveTextToCSV._get_csv_path()
      • Subpackages
        • pe.callback.common package
          • Submodules
            • pe.callback.common.compute_fid module
            • pe.callback.common.compute_precision_recall module
            • pe.callback.common.save_checkpoints module
        • pe.callback.image package
          • Subpackages
            • pe.callback.image.dpimagebench_lib package
          • Submodules
            • pe.callback.image.dpimagebench_classify_images module
            • pe.callback.image.sample_images module
            • pe.callback.image.save_all_images module
        • pe.callback.text package
          • Submodules
            • pe.callback.text.save_text_to_csv module
      • Submodules
        • pe.callback.callback module
          • Callback
            • Callback.__call__()
    • pe.constant package
      • Submodules
        • pe.constant.data module
          • CLEAN_HISTOGRAM_COLUMN_NAME
          • CLIENT_ID_COLUMN_NAME
          • DP_HISTOGRAM_COLUMN_NAME
          • EMBEDDING_COLUMN_NAME
          • FROM_LAST_FLAG_COLUMN_NAME
          • HISTOGRAM_NEAREST_NEIGHBORS_VOTING_IDS_COLUMN_NAME
          • IMAGE_DATA_COLUMN_NAME
          • IMAGE_MODEL_LABEL_COLUMN_NAME
          • IMAGE_PROMPT_COLUMN_NAME
          • LABEL_ID_COLUMN_NAME
          • LLM_PARAMETERS_COLUMN_NAME
          • LLM_REQUEST_MESSAGES_COLUMN_NAME
          • LOOKAHEAD_EMBEDDING_COLUMN_NAME
          • PARENT_SYN_DATA_INDEX_COLUMN_NAME
          • POST_PROCESSED_DP_HISTOGRAM_COLUMN_NAME
          • TEXT_DATA_COLUMN_NAME
          • VARIATION_API_FOLD_ID_COLUMN_NAME
    • pe.data package
      • Camelyon17
        • Camelyon17.__init__()
      • Cat
        • Cat.URL
        • Cat.__init__()
        • Cat._download()
        • Cat._read_data()
      • CelebA
        • CelebA.__init__()
      • Cifar10
        • Cifar10.__init__()
      • Data
        • Data.__init__()
        • Data.concat()
        • Data.filter()
        • Data.filter_label_id()
        • Data.load_checkpoint()
        • Data.merge()
        • Data.random_split()
        • Data.random_truncate()
        • Data.reset_index()
        • Data.save_checkpoint()
        • Data.set_label_id()
        • Data.split_by_client()
        • Data.split_by_index()
        • Data.truncate()
      • DigiFace1M
        • DigiFace1M.__init__()
        • DigiFace1M._read_and_process_image()
      • ImageNet
        • ImageNet.__init__()
      • MNIST
        • MNIST.__init__()
      • OpenReview
        • OpenReview.DOWNLOAD_INFO_DICT
        • OpenReview.__init__()
        • OpenReview._download()
      • PubMed
        • PubMed.DOWNLOAD_INFO_DICT
        • PubMed.__init__()
        • PubMed._download()
      • TextCSV
        • TextCSV.__init__()
      • Yelp
        • Yelp.DOWNLOAD_INFO_DICT
        • Yelp.__init__()
        • Yelp._download()
      • load_image_folder()
      • Subpackages
        • pe.data.image package
          • Submodules
            • pe.data.image.camelyon17 module
            • pe.data.image.cat module
            • pe.data.image.celeba module
            • pe.data.image.cifar10 module
            • pe.data.image.digiface1m module
            • pe.data.image.image module
            • pe.data.image.imagenet module
            • pe.data.image.mnist module
        • pe.data.text package
          • Submodules
            • pe.data.text.openreview module
            • pe.data.text.pubmed module
            • pe.data.text.text_csv module
            • pe.data.text.yelp module
      • Submodules
        • pe.data.data module
          • Data
            • Data.__init__()
            • Data.concat()
            • Data.filter()
            • Data.filter_label_id()
            • Data.load_checkpoint()
            • Data.merge()
            • Data.random_split()
            • Data.random_truncate()
            • Data.reset_index()
            • Data.save_checkpoint()
            • Data.set_label_id()
            • Data.split_by_client()
            • Data.split_by_index()
            • Data.truncate()
    • pe.dp package
      • DP
        • DP.add_noise()
        • DP.set_epsilon_and_delta()
      • Gaussian
        • Gaussian.add_noise()
        • Gaussian.set_epsilon_and_delta()
      • Submodules
        • pe.dp.dp module
          • DP
            • DP.add_noise()
            • DP.set_epsilon_and_delta()
        • pe.dp.gaussian module
          • Gaussian
            • Gaussian.add_noise()
            • Gaussian.set_epsilon_and_delta()
          • compute_epsilon()
          • delta_Gaussian()
          • eps_Gaussian()
          • get_noise_multiplier()
    • pe.embedding package
      • Embedding
        • Embedding.column_name
        • Embedding.compute_embedding()
        • Embedding.filter_uncomputed_rows()
        • Embedding.merge_computed_rows()
      • FLDInception
        • FLDInception.__init__()
        • FLDInception.compute_embedding()
      • Inception
        • Inception.__init__()
        • Inception.compute_embedding()
      • SentenceTransformer
        • SentenceTransformer.__init__()
        • SentenceTransformer.column_name
        • SentenceTransformer.compute_embedding()
      • Subpackages
        • pe.embedding.image package
          • Submodules
            • pe.embedding.image.fld_inception module
            • pe.embedding.image.inception module
        • pe.embedding.text package
          • Submodules
            • pe.embedding.text.sentence_transformer module
      • Submodules
        • pe.embedding.embedding module
          • Embedding
            • Embedding.column_name
            • Embedding.compute_embedding()
            • Embedding.filter_uncomputed_rows()
            • Embedding.merge_computed_rows()
    • pe.histogram package
      • Histogram
        • Histogram.compute_histogram()
      • NearestNeighbors
        • NearestNeighbors.__init__()
        • NearestNeighbors._compute_lookahead_embedding()
        • NearestNeighbors._log_lookahead()
        • NearestNeighbors._log_voting_details()
        • NearestNeighbors.compute_histogram()
      • Subpackages
        • pe.histogram.nearest_neighbor_backend package
          • Submodules
            • pe.histogram.nearest_neighbor_backend.auto module
            • pe.histogram.nearest_neighbor_backend.faiss module
            • pe.histogram.nearest_neighbor_backend.sklearn module
      • Submodules
        • pe.histogram.histogram module
          • Histogram
            • Histogram.compute_histogram()
        • pe.histogram.nearest_neighbors module
          • NearestNeighbors
            • NearestNeighbors.__init__()
            • NearestNeighbors._compute_lookahead_embedding()
            • NearestNeighbors._log_lookahead()
            • NearestNeighbors._log_voting_details()
            • NearestNeighbors.compute_histogram()
    • pe.llm package
      • AzureOpenAILLM
        • AzureOpenAILLM.__init__()
        • AzureOpenAILLM._get_environment_variable()
        • AzureOpenAILLM._get_response_for_one_request()
        • AzureOpenAILLM.generation_arg_map
        • AzureOpenAILLM.get_responses()
      • HuggingfaceLLM
        • HuggingfaceLLM.__init__()
        • HuggingfaceLLM._get_conv_template()
        • HuggingfaceLLM._get_prompt()
        • HuggingfaceLLM._get_responses()
        • HuggingfaceLLM.generation_arg_map
        • HuggingfaceLLM.get_responses()
      • LLM
        • LLM.generation_arg_map
        • LLM.get_generation_args()
        • LLM.get_responses()
      • OpenAILLM
        • OpenAILLM.__init__()
        • OpenAILLM._get_environment_variable()
        • OpenAILLM._get_response_for_one_request()
        • OpenAILLM.get_responses()
      • Request
      • Subpackages
        • pe.llm.huggingface package
          • Subpackages
            • pe.llm.huggingface.register_fastchat package
          • Submodules
            • pe.llm.huggingface.huggingface module
      • Submodules
        • pe.llm.azure_openai module
          • AzureOpenAILLM
            • AzureOpenAILLM.__init__()
            • AzureOpenAILLM._get_environment_variable()
            • AzureOpenAILLM._get_response_for_one_request()
            • AzureOpenAILLM.generation_arg_map
            • AzureOpenAILLM.get_responses()
        • pe.llm.llm module
          • LLM
            • LLM.generation_arg_map
            • LLM.get_generation_args()
            • LLM.get_responses()
        • pe.llm.openai module
          • OpenAILLM
            • OpenAILLM.__init__()
            • OpenAILLM._get_environment_variable()
            • OpenAILLM._get_response_for_one_request()
            • OpenAILLM.get_responses()
        • pe.llm.request module
          • Request
    • pe.logger package
      • CSVPrint
        • CSVPrint.__init__()
        • CSVPrint._clear_logs()
        • CSVPrint._flush()
        • CSVPrint._get_log_path()
        • CSVPrint._log_float()
        • CSVPrint.clean_up()
        • CSVPrint.log()
      • ImageFile
        • ImageFile.__init__()
        • ImageFile._get_image_path()
        • ImageFile._log_image()
        • ImageFile._log_image_list()
        • ImageFile.log()
      • LogPrint
        • LogPrint.__init__()
        • LogPrint.log()
      • Logger
        • Logger.clean_up()
        • Logger.log()
      • MatplotlibPDF
        • MatplotlibPDF.__init__()
        • MatplotlibPDF._get_pdf_path()
        • MatplotlibPDF.log()
      • Submodules
        • pe.logger.csv_print module
          • CSVPrint
            • CSVPrint.__init__()
            • CSVPrint._clear_logs()
            • CSVPrint._flush()
            • CSVPrint._get_log_path()
            • CSVPrint._log_float()
            • CSVPrint.clean_up()
            • CSVPrint.log()
        • pe.logger.image_file module
          • ImageFile
            • ImageFile.__init__()
            • ImageFile._get_image_path()
            • ImageFile._log_image()
            • ImageFile._log_image_list()
            • ImageFile.log()
        • pe.logger.log_print module
          • LogPrint
            • LogPrint.__init__()
            • LogPrint.log()
        • pe.logger.logger module
          • Logger
            • Logger.clean_up()
            • Logger.log()
        • pe.logger.matplotlib_pdf module
          • MatplotlibPDF
            • MatplotlibPDF.__init__()
            • MatplotlibPDF._get_pdf_path()
            • MatplotlibPDF.log()
    • pe.logging package
      • execution_logger
      • setup_logging()
    • pe.metric_item package
      • FloatListMetricItem
      • FloatMetricItem
      • ImageListMetricItem
        • ImageListMetricItem.__init__()
        • ImageListMetricItem.num_images_per_row
      • ImageMetricItem
      • MatplotlibMetricItem
        • MatplotlibMetricItem.clean_up()
      • MetricItem
        • MetricItem.__init__()
        • MetricItem.clean_up()
        • MetricItem.name
        • MetricItem.value
      • metric_scope
    • pe.population package
      • PEPopulation
        • PEPopulation.__init__()
        • PEPopulation._log_histogram()
        • PEPopulation._post_process_histogram()
        • PEPopulation._select_data()
        • PEPopulation.initial()
        • PEPopulation.next()
      • Population
        • Population.initial()
        • Population.next()
      • Submodules
        • pe.population.pe_population module
          • PEPopulation
            • PEPopulation.__init__()
            • PEPopulation._log_histogram()
            • PEPopulation._post_process_histogram()
            • PEPopulation._select_data()
            • PEPopulation.initial()
            • PEPopulation.next()
        • pe.population.population module
          • Population
            • Population.initial()
            • Population.next()
    • pe.runner package
      • PE
        • PE.__init__()
        • PE._clean_up_loggers()
        • PE._get_num_samples_per_label_id()
        • PE._log_metrics()
        • PE.evaluate()
        • PE.load_checkpoint()
        • PE.run()
      • Submodules
        • pe.runner.pe module
          • PE
            • PE.__init__()
            • PE._clean_up_loggers()
            • PE._get_num_samples_per_label_id()
            • PE._log_metrics()
            • PE.evaluate()
            • PE.load_checkpoint()
            • PE.run()
    • pe.util package
      • Submodules
        • pe.util.download module
          • download()