src package

Subpackages

• src.model package
  • Submodules
    • src.model.concept_history module
      • get_initial_concepts()
      • update_concept_history()
    • src.model.explained_model module
      • ExplainedModel
    • src.model.image_model module
      • ImageModel
    • src.model.language_model module
      • LanguageModel
    • src.model.model module
      • Model
      • gpu_inference_wrapper()
  • Module contents
• src.prompts package
  • Submodules
    • src.prompts.prompt_utils module
      • concept_image_prompt()
      • generate_prompt()
  • Module contents

Submodules

• src.activation_sampling module
  • ActivationSampler
    • ActivationSampler.sample_control_activations()
• src.calculate_control_activations module
  • main()
• src.config module
  • ConceptHistoryConfig
    • ConceptHistoryConfig.n_best_concepts
    • ConceptHistoryConfig.n_random_concepts
    • ConceptHistoryConfig.n_best_concepts
    • ConceptHistoryConfig.n_random_concepts
  • HistoryManagingConfig
    • HistoryManagingConfig.save_images
    • HistoryManagingConfig.save_histories
    • HistoryManagingConfig.save_directory
    • HistoryManagingConfig.save_directory
    • HistoryManagingConfig.save_histories
    • HistoryManagingConfig.save_images
  • ImageGenerationConfig
    • ImageGenerationConfig.n_images
    • ImageGenerationConfig.prompt_text
    • ImageGenerationConfig.n_images
    • ImageGenerationConfig.prompt_text
  • LoadConfig
    • LoadConfig.language_model_id
    • LoadConfig.text_to_image_model_id
    • LoadConfig.explained_model_id
    • LoadConfig.language_model_kwargs
    • LoadConfig.text_to_image_model_kwargs
    • LoadConfig.explained_model_kwargs
    • LoadConfig.model_swapping
    • LoadConfig.explained_model_id
    • LoadConfig.explained_model_kwargs
    • LoadConfig.language_model_id
    • LoadConfig.language_model_kwargs
    • LoadConfig.model_swapping
    • LoadConfig.text_to_image_model_id
    • LoadConfig.text_to_image_model_kwargs
• src.history_managing module
  • format_as_json_string()
  • format_best_concepts_history()
  • format_concept_history()
  • save_images_from_iteration()
  • save_llm_history()
  • save_pipeline_parameters()
• src.image_processing module
  • ConditionalNormalize
  • save_images_from_iteration()
  • transform_images()
• src.main module
  • main()
• src.pipeline module
  • Pipeline
    • Pipeline.run_pipeline()
• src.scoring module
  • Metric
    • Metric.AUC
    • Metric.MAD
    • Metric.AVG_ACTIVATION
    • Metric.AUC
    • Metric.AVG_ACTIVATION
    • Metric.MAD
  • calculate_metric()

Module contents