Index

_

__getitem__() (datagenerator.BaseFeaturesDataset method) (datagenerator.boolean.BooleanAndDataset method) (datagenerator.boolean.BooleanDataset method) (datagenerator.boolean.BooleanOrDataset method) (datagenerator.BooleanAndDataset method) (datagenerator.BooleanDataset method) (datagenerator.BooleanOrDataset method) (datagenerator.data_generation.BaseFeaturesDataset method) (datagenerator.image_generation.ImageDataset method) (datagenerator.ImageDataset method) (datagenerator.pertinent_negatives.PertinentNegativesDataset method) (datagenerator.PertinentNegativesDataset method) (datagenerator.shattered_grad.ShatteredGradientsDataset method) (datagenerator.ShatteredGradientsDataset method) (datagenerator.text_datasets.TextTriggerDataset method) (datagenerator.TextTriggerDataset method) (datagenerator.uncertainty_aware.UncertaintyAwareDataset method) (datagenerator.UncertaintyAwareDataset method) __len__() (datagenerator.BaseFeaturesDataset method) (datagenerator.data_generation.BaseFeaturesDataset method) (datagenerator.image_generation.ImageDataset method) (datagenerator.ImageDataset method) (datagenerator.text_datasets.TextTriggerDataset method) (datagenerator.TextTriggerDataset method) __post_init__() (pipeline.Example method) (pipeline.results.Example method) __repr__() (model.boolean.ParseTree method) _all_models_explanation_attributes() (pipeline.BasePipeline method) (pipeline.pipeline.BasePipeline method) _and_or_weights() (model.boolean.PropFormulaNN method) (model.PropFormulaNN method) _apply_attribution_method() (pipeline.BasePipeline method) (pipeline.pipeline.BasePipeline method) _atom_layer() (model.boolean.PropFormulaNN method) (model.PropFormulaNN method) _attr_time_summing() (pipeline.Results method) (pipeline.results.Results method) _binary_operator_block() (model.boolean.PropFormulaNN method) (model.PropFormulaNN method) _colors_rgba (datagenerator.foregrounds.ForegroundGenerator attribute) _count_binary_operators() (model.boolean.PropFormulaNN method) (model.PropFormulaNN method) _create_ground_truth_baseline() (datagenerator.pertinent_negatives.PertinentNegativesDataset method) (datagenerator.PertinentNegativesDataset method) _create_identity_trees() (model.boolean.ParseTree method) _create_layer_weights() (model.boolean_and.BooleanAndNN method) (model.boolean_or.BooleanOrNN method) (model.BooleanAndNN method) (model.BooleanOrNN method) (model.conflicting.ConflictingFeaturesNN method) (model.ConflictingFeaturesNN method) (model.continuous.ContinuousFeaturesNN method) (model.ContinuousFeaturesNN method) (model.InteractingFeaturesNN method) (model.interaction_features.InteractingFeaturesNN method) (model.pertinent_negative.PertinentNN method) (model.PertinentNN method) _create_weights() (datagenerator.uncertainty_aware.UncertaintyAwareDataset method) (datagenerator.UncertaintyAwareDataset method) _data_filename (datagenerator.backgrounds.BackgroundGenerator attribute) _data_folder (datagenerator.backgrounds.BackgroundGenerator attribute) (datagenerator.foregrounds.DinosaurShapeGenerator attribute) _data_path (datagenerator.foregrounds.ForegroundGenerator attribute) _data_url (datagenerator.backgrounds.BackgroundGenerator attribute) _default_activation_function() (datagenerator.shattered_grad.ShatteredGradientsDataset method) (datagenerator.ShatteredGradientsDataset method) (model.shattered_gradients.ShatteredGradientsNN method) (model.ShatteredGradientsNN method) _fill_height_gaps() (model.boolean.ParseTree method) _flatten_cat_features() (in module metrics) (in module metrics.metrics_wrapper) _generate_default_metrics() (in module datagenerator.text_datasets) _generate_default_weights() (datagenerator.shattered_grad.ShatteredGradientsDataset method) (datagenerator.ShatteredGradientsDataset method) _get_cancellation_attributions() (datagenerator.conflicting.ConflictingDataset method) (datagenerator.ConflictingDataset method) _get_cancellation_samples() (datagenerator.conflicting.ConflictingDataset method) (datagenerator.ConflictingDataset method) _get_cancellations() (datagenerator.conflicting.ConflictingDataset method) (datagenerator.ConflictingDataset method) _get_default_distribution_narrow() (datagenerator.shattered_grad.ShatteredGradientsDataset method) (datagenerator.ShatteredGradientsDataset method) _get_default_distribution_wide() (datagenerator.shattered_grad.ShatteredGradientsDataset method) (datagenerator.ShatteredGradientsDataset method) _get_flat_weights() (datagenerator.interacting_features.InteractingFeatureDataset method) (datagenerator.InteractingFeatureDataset method) _get_height() (model.boolean.ParseTree method) _get_height_layers() (model.boolean.PropFormulaNN method) (model.PropFormulaNN method) _get_new_weighted_samples() (datagenerator.pertinent_negatives.PertinentNegativesDataset method) (datagenerator.PertinentNegativesDataset method)

_get_weight_scale() (datagenerator.shattered_grad.ShatteredGradientsDataset method) (datagenerator.ShatteredGradientsDataset method) _identity_weights() (model.boolean.PropFormulaNN method) (model.PropFormulaNN method) _init_attr() (pipeline.BasePipeline method) (pipeline.pipeline.BasePipeline method) _init_binary_operator_layers() (model.boolean.PropFormulaNN method) (model.PropFormulaNN method) _init_data() (pipeline.Experiment method) (pipeline.experiment.Experiment method) _init_data_params() (pipeline.Experiment method) (pipeline.experiment.Experiment method) _init_layers() (model.boolean.PropFormulaNN method) (model.PropFormulaNN method) _init_method_seeds() (pipeline.Experiment method) (pipeline.experiment.Experiment method) _init_methods() (pipeline.Experiment method) (pipeline.experiment.Experiment method) _init_metrics() (pipeline.Experiment method) (pipeline.experiment.Experiment method) _init_models() (pipeline.Experiment method) (pipeline.experiment.Experiment method) _init_noise_parameters() (datagenerator.BaseFeaturesDataset method) (datagenerator.data_generation.BaseFeaturesDataset method) _init_none_metric() (pipeline.Pipeline method) (pipeline.pipeline.Pipeline method) _init_samples() (datagenerator.BaseFeaturesDataset method) (datagenerator.data_generation.BaseFeaturesDataset method) _init_seeds() (pipeline.Experiment method) (pipeline.experiment.Experiment method) _init_unary_operator_layer() (model.boolean.PropFormulaNN method) (model.PropFormulaNN method) _initialize_cancellation_features() (datagenerator.conflicting.ConflictingDataset method) (datagenerator.ConflictingDataset method) _initialize_discontinuity_ratios() (datagenerator.shattered_grad.ShatteredGradientsDataset method) (datagenerator.ShatteredGradientsDataset method) _initialize_samples_labels() (datagenerator.boolean.BooleanDataset method) (datagenerator.BooleanDataset method) _initialize_samples_narrow_wide() (datagenerator.shattered_grad.ShatteredGradientsDataset method) (datagenerator.ShatteredGradientsDataset method) _initialize_weights() (datagenerator.data_generation.WeightedFeaturesDataset method) (datagenerator.WeightedFeaturesDataset method) _initialize_with_narrow_distribution() (datagenerator.shattered_grad.ShatteredGradientsDataset method) (datagenerator.ShatteredGradientsDataset method) _initialize_with_narrow_wide_distributions() (datagenerator.shattered_grad.ShatteredGradientsDataset method) (datagenerator.ShatteredGradientsDataset method) _initialize_zeros_for_PN() (datagenerator.pertinent_negatives.PertinentNegativesDataset method) (datagenerator.PertinentNegativesDataset method) _intialize_pn_features() (datagenerator.pertinent_negatives.PertinentNegativesDataset method) (datagenerator.PertinentNegativesDataset method) _parse() (model.boolean.ParseTree method) _post_block_arg_num_ls() (model.boolean.PropFormulaNN method) (model.PropFormulaNN method) _prepare_backgrounds() (datagenerator.image_generation.ImageDataset method) (datagenerator.ImageDataset method) _prepare_shape_color() (datagenerator.image_generation.ImageDataset method) (datagenerator.image_generation.ImbalancedImageDataset method) (datagenerator.ImageDataset method) (datagenerator.ImbalancedImageDataset method) _prepare_shapes() (datagenerator.image_generation.ImageDataset method) (datagenerator.ImageDataset method) _re_label() (datagenerator.image_generation.ImageDataset method) (datagenerator.ImageDataset method) _reformat_pn_weight() (model.pertinent_negative.PertinentNN method) (model.PertinentNN method) _reformat_replacements() (in module metrics) (in module metrics.metrics_wrapper) _single_explanation_attribute() (pipeline.BasePipeline method) (pipeline.pipeline.BasePipeline method) _store_top_n() (pipeline.BasePipeline method) (pipeline.pipeline.BasePipeline method) _url (datagenerator.foregrounds.DinosaurShapeGenerator attribute) _validate_and_reformat_inputs() (in module model.generic) _validate_background_size() (datagenerator.backgrounds.BackgroundGenerator method) _validate_imbalance() (datagenerator.image_generation.ImbalancedImageDataset method) (datagenerator.ImbalancedImageDataset method) _validate_input_gen_arguments() (in module methods) (in module methods.methods_wrapper) _validate_inputs() (datagenerator.BaseFeaturesDataset method) (datagenerator.data_generation.BaseFeaturesDataset method) (model.InteractingFeaturesNN method) (model.interaction_features.InteractingFeaturesNN method) _validate_metric_gen_arguments() (in module metrics) (in module metrics.metrics_wrapper) _validate_n_variants() (datagenerator.image_generation.ImageDataset method) (datagenerator.ImageDataset method) _validate_wrapper_inputs() (in module methods) (in module methods.methods_wrapper) (in module metrics) (in module metrics.metrics_wrapper) _verify_metric() (pipeline.Experiment method) (pipeline.experiment.Experiment method)

A

add_feature_mask (in module fine_tune) add_trigger_word() (in module common_lib) append() (pipeline.Results method) (pipeline.results.Results method) apply_color_fill() (datagenerator.foregrounds.ForegroundGenerator method) args (in module fine_tune) atoms (datagenerator.boolean.BooleanAndDataset attribute) (datagenerator.boolean.BooleanDataset attribute), [1] (datagenerator.boolean.BooleanOrDataset attribute) (datagenerator.BooleanAndDataset attribute) (datagenerator.BooleanDataset attribute), [1] (datagenerator.BooleanOrDataset attribute) (model.boolean.PropFormulaNN attribute), [1] (model.PropFormulaNN attribute), [1]

attribute (pipeline.Example attribute) (pipeline.results.Example attribute) AutoTrainer (class in trainer) (class in trainer.trainer)

B

back_gen (datagenerator.image_generation.ImageBuilder attribute), [1] background_names (datagenerator.backgrounds.BackgroundGenerator attribute), [1] background_size (datagenerator.backgrounds.BackgroundGenerator attribute), [1] (datagenerator.image_generation.ImageBuilder attribute), [1] (datagenerator.image_generation.ImageDataset attribute) (datagenerator.ImageDataset attribute) BackgroundGenerator (class in datagenerator.backgrounds) backgrounds (datagenerator.image_generation.ImageDataset attribute), [1] (datagenerator.ImageDataset attribute), [1] balanced_dataset (in module datagenerator.image_generation) BalancedImageDataset (class in datagenerator) (class in datagenerator.image_generation) BaseFeaturesDataset (class in datagenerator) (class in datagenerator.data_generation) baselines (datagenerator.text_datasets.TextTriggerDataset attribute) (datagenerator.TextTriggerDataset attribute) BasePipeline (class in pipeline) (class in pipeline.pipeline)

BaseTextDataset (class in datagenerator.text_datasets) batch_size (pipeline.BasePipeline attribute), [1] (pipeline.pipeline.BasePipeline attribute), [1] best_checkpoint (in module fine_tune) bg_labels (datagenerator.image_generation.ImageDataset attribute), [1] (datagenerator.ImageDataset attribute), [1] BooleanAndDataset (class in datagenerator) (class in datagenerator.boolean) BooleanAndNN (class in model) (class in model.boolean_and) BooleanDataset (class in datagenerator) (class in datagenerator.boolean) BooleanNotNN (class in model) (class in model.boolean_not) BooleanOrDataset (class in datagenerator) (class in datagenerator.boolean) BooleanOrNN (class in model) (class in model.boolean_or)

C

calculate_ngon_vertices() (datagenerator.foregrounds.GeometricShapeGenerator method) cancellation_attributions (datagenerator.conflicting.ConflictingDataset attribute), [1] (datagenerator.ConflictingDataset attribute), [1] cancellation_features (datagenerator.conflicting.ConflictingDataset attribute), [1] (datagenerator.ConflictingDataset attribute), [1] cancellation_likelihood (datagenerator.conflicting.ConflictingDataset attribute), [1] (datagenerator.ConflictingDataset attribute), [1] cancellation_outcomes (datagenerator.conflicting.ConflictingDataset attribute), [1] (datagenerator.ConflictingDataset attribute), [1] cancellation_samples (datagenerator.conflicting.ConflictingDataset attribute), [1] (datagenerator.ConflictingDataset attribute), [1] cat_features (datagenerator.BaseFeaturesDataset attribute), [1] (datagenerator.boolean.BooleanDataset attribute) (datagenerator.BooleanDataset attribute) (datagenerator.conflicting.ConflictingDataset attribute), [1] (datagenerator.ConflictingDataset attribute), [1] (datagenerator.data_generation.BaseFeaturesDataset attribute), [1] (datagenerator.interacting_features.InteractingFeatureDataset attribute) (datagenerator.InteractingFeatureDataset attribute) (datagenerator.pertinent_negatives.PertinentNegativesDataset attribute), [1] (datagenerator.PertinentNegativesDataset attribute), [1] CHAT_TEMPLATE (in module common_lib) clean_dino_name_from_URL() (datagenerator.foregrounds.DinosaurShapeGenerator method) collate_fn (datagenerator.text_datasets.TextTriggerDataset property) (datagenerator.TextTriggerDataset property) color (datagenerator.image_generation.ImageBuilder attribute), [1]

common_features (datagenerator.uncertainty_aware.UncertaintyAwareDataset attribute) (datagenerator.UncertaintyAwareDataset attribute) common_lib module configure_optimizers() (trainer.AutoTrainer method) (trainer.trainer.AutoTrainer method) ConflictingDataset (class in datagenerator) (class in datagenerator.conflicting) ConflictingFeaturesNN (class in model) (class in model.conflicting) context (pipeline.Example attribute) (pipeline.results.Example attribute) continuous_dim (in module model.conflicting) ContinuousFeaturesNN (class in model) (class in model.continuous) contour_thickness (datagenerator.image_generation.ImageDataset attribute), [1] (datagenerator.ImageDataset attribute), [1] create_baselines() (datagenerator.boolean.BooleanAndDataset method) (datagenerator.boolean.BooleanOrDataset method) (datagenerator.BooleanAndDataset method) (datagenerator.BooleanOrDataset method) create_eval_dataset() (in module common_lib) create_ground_truth() (datagenerator.boolean.BooleanAndDataset method) (datagenerator.boolean.BooleanOrDataset method) (datagenerator.BooleanAndDataset method) (datagenerator.BooleanOrDataset method) create_train_dataset() (in module common_lib)

D

data (in module datagenerator) (in module datagenerator.boolean) (in module datagenerator.conflicting) (in module datagenerator.data_generation) (in module datagenerator.interacting_features) (in module datagenerator.pertinent_negatives) (in module datagenerator.shattered_grad) (in module metrics) (in module metrics.metrics_wrapper) (in module model.interaction_features) (in module model.pertinent_negative) (pipeline.Experiment attribute), [1] (pipeline.experiment.Experiment attribute), [1] (pipeline.Results attribute) (pipeline.Results property) (pipeline.results.Results attribute) (pipeline.results.Results property) data_gen (in module trainer) (in module trainer.trainer) data_params (pipeline.Experiment attribute), [1] (pipeline.experiment.Experiment attribute), [1] datagenerator module datagenerator.backgrounds module datagenerator.boolean module datagenerator.conflicting module datagenerator.data_generation module datagenerator.foregrounds module datagenerator.image_generation module datagenerator.interacting_features module datagenerator.pertinent_negatives module datagenerator.shattered_grad module datagenerator.text_datasets module datagenerator.uncertainty_aware module

datas (pipeline.Pipeline attribute), [1] (pipeline.pipeline.Pipeline attribute), [1] dataset (in module datagenerator.text_datasets) (in module fine_tune) dataset_types (in module fine_tune) default_attribute_input_generator() (in module methods) (in module methods.methods_wrapper) default_background_imagefile (datagenerator.image_generation.ImageBuilder attribute), [1] default_metric (datagenerator.BaseFeaturesDataset property) (datagenerator.boolean.BooleanAndDataset property) (datagenerator.boolean.BooleanDataset property) (datagenerator.boolean.BooleanOrDataset property) (datagenerator.BooleanAndDataset property) (datagenerator.BooleanDataset property) (datagenerator.BooleanOrDataset property) (datagenerator.data_generation.BaseFeaturesDataset property) (datagenerator.data_generation.WeightedFeaturesDataset property) (datagenerator.image_generation.ImageDataset property) (datagenerator.ImageDataset property) (datagenerator.shattered_grad.ShatteredGradientsDataset property) (datagenerator.ShatteredGradientsDataset property) (datagenerator.text_datasets.TextTriggerDataset property) (datagenerator.TextTriggerDataset property) (datagenerator.uncertainty_aware.UncertaintyAwareDataset property) (datagenerator.UncertaintyAwareDataset property) (datagenerator.WeightedFeaturesDataset property) default_metric_input_generator() (in module metrics) (in module metrics.metrics_wrapper) default_target (pipeline.BasePipeline attribute), [1] (pipeline.pipeline.BasePipeline attribute), [1] dims (model.boolean_and.BooleanAndNN attribute) (model.boolean_or.BooleanOrNN attribute) (model.BooleanAndNN attribute) (model.BooleanOrNN attribute) dino_dict (datagenerator.foregrounds.DinosaurShapeGenerator attribute), [1] dino_image_metadata() (datagenerator.foregrounds.DinosaurShapeGenerator method) dino_meta_data (datagenerator.foregrounds.DinosaurShapeGenerator attribute), [1] DinosaurShapeGenerator (class in datagenerator.foregrounds) distribution (datagenerator.BaseFeaturesDataset attribute) (datagenerator.data_generation.BaseFeaturesDataset attribute) do_training_sft() (in module common_lib) download() (datagenerator.backgrounds.BackgroundGenerator method) DynamicNN (class in model) (class in model.dynamic)

E

Example (class in pipeline) (class in pipeline.results) example_type (pipeline.Example attribute) (pipeline.results.Example attribute) examples (pipeline.Results attribute), [1] (pipeline.results.Results attribute), [1] exp_name (pipeline.Experiment attribute) (pipeline.experiment.Experiment attribute) Experiment (class in pipeline) (class in pipeline.experiment)

ExperimentPipeline (class in pipeline) (class in pipeline.pipeline) experiments (pipeline.ExperimentPipeline attribute), [1] (pipeline.pipeline.ExperimentPipeline attribute), [1] explanation_attribute() (pipeline.ExperimentPipeline method) (pipeline.Pipeline method) (pipeline.pipeline.ExperimentPipeline method) (pipeline.pipeline.Pipeline method)

F

feature (in module model.boolean_and) (in module model.boolean_or) feature_inputs (pipeline.Example attribute) (pipeline.results.Example attribute) features (datagenerator.BaseFeaturesDataset attribute) (datagenerator.conflicting.ConflictingDataset attribute) (datagenerator.ConflictingDataset attribute) (datagenerator.data_generation.BaseFeaturesDataset attribute) (datagenerator.data_generation.WeightedFeaturesDataset attribute) (datagenerator.pertinent_negatives.PertinentNegativesDataset attribute), [1] (datagenerator.PertinentNegativesDataset attribute), [1] (datagenerator.uncertainty_aware.UncertaintyAwareDataset attribute) (datagenerator.UncertaintyAwareDataset attribute) (datagenerator.WeightedFeaturesDataset attribute)

fg_colors (datagenerator.image_generation.ImageDataset attribute), [1] (datagenerator.ImageDataset attribute), [1] fg_shapes (datagenerator.image_generation.ImageDataset attribute), [1] (datagenerator.ImageDataset attribute), [1] fine_tune module ForegroundGenerator (class in datagenerator.foregrounds) formula (datagenerator.boolean.BooleanAndDataset attribute) (datagenerator.boolean.BooleanDataset attribute), [1] (datagenerator.boolean.BooleanOrDataset attribute) (datagenerator.BooleanAndDataset attribute) (datagenerator.BooleanDataset attribute), [1] (datagenerator.BooleanOrDataset attribute) forward() (trainer.AutoTrainer method) (trainer.trainer.AutoTrainer method)

G

generate_csv() (in module datagenerator) (in module datagenerator.data_generation) generate_feature_mask() (in module common_lib) generate_layers() (in module model) (in module model.generic) generate_model() (datagenerator.BaseFeaturesDataset method) (datagenerator.boolean.BooleanAndDataset method) (datagenerator.boolean.BooleanDataset method) (datagenerator.boolean.BooleanOrDataset method) (datagenerator.BooleanAndDataset method) (datagenerator.BooleanDataset method) (datagenerator.BooleanOrDataset method) (datagenerator.conflicting.ConflictingDataset method) (datagenerator.ConflictingDataset method) (datagenerator.data_generation.BaseFeaturesDataset method) (datagenerator.data_generation.WeightedFeaturesDataset method) (datagenerator.interacting_features.InteractingFeatureDataset method) (datagenerator.InteractingFeatureDataset method) (datagenerator.pertinent_negatives.PertinentNegativesDataset method) (datagenerator.PertinentNegativesDataset method) (datagenerator.shattered_grad.ShatteredGradientsDataset method) (datagenerator.ShatteredGradientsDataset method) (datagenerator.text_datasets.TextTriggerDataset method) (datagenerator.TextTriggerDataset method) (datagenerator.uncertainty_aware.UncertaintyAwareDataset method) (datagenerator.UncertaintyAwareDataset method) (datagenerator.WeightedFeaturesDataset method) generate_sample() (datagenerator.image_generation.ImageBuilder method) generate_samples() (datagenerator.BalancedImageDataset method) (datagenerator.image_generation.BalancedImageDataset method) (datagenerator.image_generation.ImageDataset method) (datagenerator.image_generation.ImbalancedImageDataset method) (datagenerator.ImageDataset method) (datagenerator.ImbalancedImageDataset method) GenericNN (class in model) (class in model.generic) geometric_shapes() (datagenerator.foregrounds.GeometricShapeGenerator method) GeometricShapeGenerator (class in datagenerator.foregrounds) get_background() (datagenerator.backgrounds.BackgroundGenerator method)

get_background_names() (datagenerator.backgrounds.BackgroundGenerator method) get_data() (pipeline.Experiment method) (pipeline.experiment.Experiment method) get_data_path() (datagenerator.backgrounds.BackgroundGenerator method) (datagenerator.foregrounds.ForegroundGenerator method) get_height_subtrees() (model.boolean.ParseTree method) get_leafs() (model.boolean.ParseTree method) get_methods() (pipeline.Experiment method) (pipeline.experiment.Experiment method) get_metrics() (pipeline.Experiment method) (pipeline.experiment.Experiment method) get_model_responses() (in module common_lib) get_model_tokenizer() (in module common_lib) get_models() (pipeline.Experiment method) (pipeline.experiment.Experiment method) get_shape() (datagenerator.foregrounds.DinosaurShapeGenerator method) (datagenerator.foregrounds.ForegroundGenerator method) (datagenerator.foregrounds.GeometricShapeGenerator method) ground_truth (datagenerator.boolean.BooleanAndDataset attribute) (datagenerator.boolean.BooleanOrDataset attribute) (datagenerator.BooleanAndDataset attribute) (datagenerator.BooleanOrDataset attribute) (datagenerator.image_generation.ImageDataset attribute), [1] (datagenerator.ImageDataset attribute), [1] ground_truth_attribute (datagenerator.BaseFeaturesDataset attribute), [1] (datagenerator.boolean.BooleanAndDataset attribute) (datagenerator.boolean.BooleanOrDataset attribute) (datagenerator.BooleanAndDataset attribute) (datagenerator.BooleanOrDataset attribute) (datagenerator.conflicting.ConflictingDataset attribute) (datagenerator.ConflictingDataset attribute) (datagenerator.data_generation.BaseFeaturesDataset attribute), [1] (datagenerator.data_generation.WeightedFeaturesDataset attribute) (datagenerator.pertinent_negatives.PertinentNegativesDataset attribute), [1] (datagenerator.PertinentNegativesDataset attribute), [1] (datagenerator.uncertainty_aware.UncertaintyAwareDataset attribute) (datagenerator.UncertaintyAwareDataset attribute) (datagenerator.WeightedFeaturesDataset attribute) ground_truth_attributions (datagenerator.conflicting.ConflictingDataset attribute), [1] (datagenerator.ConflictingDataset attribute), [1]

H

height (model.boolean.ParseTree attribute)

I

image_builder (datagenerator.image_generation.ImageDataset attribute), [1] (datagenerator.ImageDataset attribute), [1] image_gen (datagenerator.image_generation.ImageBuilder attribute) ImageBuilder (class in datagenerator.image_generation) ImageDataset (class in datagenerator) (class in datagenerator.image_generation) imbalance (datagenerator.image_generation.ImbalancedImageDataset attribute), [1] (datagenerator.ImbalancedImageDataset attribute), [1] ImbalancedImageDataset (class in datagenerator) (class in datagenerator.image_generation)

img_rotation() (datagenerator.image_generation.ImageBuilder method) index (datagenerator.text_datasets.TextTriggerDataset attribute) (datagenerator.TextTriggerDataset attribute) inference_batch_size (in module fine_tune) interacting_features (datagenerator.interacting_features.InteractingFeatureDataset attribute), [1] (datagenerator.InteractingFeatureDataset attribute), [1] InteractingFeatureDataset (class in datagenerator) (class in datagenerator.interacting_features) InteractingFeaturesNN (class in model) (class in model.interaction_features)

K

k (in module model.boolean)

L

label_noise (datagenerator.BaseFeaturesDataset attribute) (datagenerator.data_generation.BaseFeaturesDataset attribute) (datagenerator.data_generation.WeightedFeaturesDataset attribute) (datagenerator.pertinent_negatives.PertinentNegativesDataset attribute) (datagenerator.PertinentNegativesDataset attribute) (datagenerator.WeightedFeaturesDataset attribute) label_std_dev (datagenerator.BaseFeaturesDataset attribute) (datagenerator.data_generation.BaseFeaturesDataset attribute) labels (datagenerator.BaseFeaturesDataset attribute), [1] (datagenerator.conflicting.ConflictingDataset attribute) (datagenerator.ConflictingDataset attribute) (datagenerator.data_generation.BaseFeaturesDataset attribute), [1] (datagenerator.data_generation.WeightedFeaturesDataset attribute) (datagenerator.image_generation.ImageDataset attribute), [1] (datagenerator.ImageDataset attribute), [1] (datagenerator.pertinent_negatives.PertinentNegativesDataset attribute), [1] (datagenerator.PertinentNegativesDataset attribute), [1] (datagenerator.uncertainty_aware.UncertaintyAwareDataset attribute), [1] (datagenerator.UncertaintyAwareDataset attribute), [1] (datagenerator.WeightedFeaturesDataset attribute)

load() (pipeline.Results method) (pipeline.results.Results method) load_all_dinos() (datagenerator.foregrounds.DinosaurShapeGenerator method) load_dataset() (in module datagenerator) (in module datagenerator.data_generation) load_dino_image() (datagenerator.foregrounds.DinosaurShapeGenerator method) local (in module fine_tune) loss (trainer.AutoTrainer attribute), [1] (trainer.trainer.AutoTrainer attribute), [1]

M

make_cat() (datagenerator.interacting_features.InteractingFeatureDataset method) (datagenerator.InteractingFeatureDataset method) make_circle() (datagenerator.foregrounds.GeometricShapeGenerator method) make_ellipse() (datagenerator.foregrounds.GeometricShapeGenerator method) make_ngon() (datagenerator.foregrounds.GeometricShapeGenerator method) make_rectangle() (datagenerator.foregrounds.GeometricShapeGenerator method) mask (datagenerator.uncertainty_aware.UncertaintyAwareDataset attribute) (datagenerator.UncertaintyAwareDataset attribute) max_sequence_length (datagenerator.text_datasets.TextTriggerDataset attribute) (datagenerator.TextTriggerDataset attribute) meta_data_source (datagenerator.foregrounds.DinosaurShapeGenerator attribute) method_seed (pipeline.Pipeline attribute), [1] (pipeline.pipeline.Pipeline attribute), [1] method_seeds (pipeline.Experiment attribute), [1] (pipeline.experiment.Experiment attribute), [1] methods module methods (in module pipeline) (in module pipeline.pipeline) (pipeline.Experiment attribute), [1] (pipeline.experiment.Experiment attribute), [1] (pipeline.Pipeline attribute), [1] (pipeline.pipeline.Pipeline attribute), [1] methods.methods_wrapper module metrics module metrics (pipeline.Experiment attribute), [1] (pipeline.experiment.Experiment attribute), [1] (pipeline.Pipeline attribute) (pipeline.pipeline.Pipeline attribute) (pipeline.Results attribute) (pipeline.results.Results attribute) metrics.metrics_wrapper module mlp_example (in module model.dynamic) model module model (in module methods) (in module methods.methods_wrapper) (in module model.boolean_not) (trainer.AutoTrainer attribute), [1] (trainer.trainer.AutoTrainer attribute), [1] model.boolean module model.boolean_and module model.boolean_not module model.boolean_or module model.conflicting module model.continuous module model.dynamic module

model.generic module model.interaction_features module model.pertinent_negative module model.shattered_gradients module model.uncertainty_model module model_name (datagenerator.text_datasets.TextTriggerDataset attribute), [1] (datagenerator.TextTriggerDataset attribute), [1] (in module fine_tune) models (pipeline.Experiment attribute), [1] (pipeline.experiment.Experiment attribute), [1] (pipeline.Pipeline attribute), [1] (pipeline.pipeline.Pipeline attribute), [1] module common_lib datagenerator datagenerator.backgrounds datagenerator.boolean datagenerator.conflicting datagenerator.data_generation datagenerator.foregrounds datagenerator.image_generation datagenerator.interacting_features datagenerator.pertinent_negatives datagenerator.shattered_grad datagenerator.text_datasets datagenerator.uncertainty_aware fine_tune methods methods.methods_wrapper metrics metrics.metrics_wrapper model model.boolean model.boolean_and model.boolean_not model.boolean_or model.conflicting model.continuous model.dynamic model.generic model.interaction_features model.pertinent_negative model.shattered_gradients model.uncertainty_model pipeline pipeline.experiment pipeline.pipeline pipeline.results trainer trainer.trainer

N

n_examples (pipeline.BasePipeline attribute), [1] (pipeline.pipeline.BasePipeline attribute), [1] n_features (datagenerator.BaseFeaturesDataset attribute) (datagenerator.boolean.BooleanAndDataset attribute) (datagenerator.boolean.BooleanOrDataset attribute) (datagenerator.BooleanAndDataset attribute) (datagenerator.BooleanOrDataset attribute) (datagenerator.data_generation.BaseFeaturesDataset attribute) (datagenerator.interacting_features.InteractingFeatureDataset attribute) (datagenerator.InteractingFeatureDataset attribute) (model.boolean_and.BooleanAndNN attribute) (model.boolean_or.BooleanOrNN attribute) (model.BooleanAndNN attribute) (model.BooleanOrNN attribute) n_samples (datagenerator.BaseFeaturesDataset attribute) (datagenerator.boolean.BooleanAndDataset attribute) (datagenerator.boolean.BooleanDataset attribute) (datagenerator.boolean.BooleanOrDataset attribute) (datagenerator.BooleanAndDataset attribute) (datagenerator.BooleanDataset attribute) (datagenerator.BooleanOrDataset attribute) (datagenerator.data_generation.BaseFeaturesDataset attribute) (datagenerator.interacting_features.InteractingFeatureDataset attribute) (datagenerator.InteractingFeatureDataset attribute)

n_variants (datagenerator.image_generation.ImageDataset attribute), [1] (datagenerator.ImageDataset attribute), [1] name (datagenerator.BaseFeaturesDataset attribute) (datagenerator.boolean.BooleanDataset attribute) (datagenerator.BooleanDataset attribute) (datagenerator.data_generation.BaseFeaturesDataset attribute) node (model.boolean.ParseTree attribute)

O

optimizer (trainer.AutoTrainer attribute), [1] (trainer.trainer.AutoTrainer attribute), [1] optimizer_params (trainer.AutoTrainer attribute), [1] (trainer.trainer.AutoTrainer attribute), [1]

output_dir (in module fine_tune) overlay_pos() (datagenerator.image_generation.ImageBuilder method) overlay_scale (datagenerator.image_generation.ImageBuilder attribute), [1] (datagenerator.image_generation.ImageDataset attribute) (datagenerator.ImageDataset attribute)

P

parse_tree (model.boolean.PropFormulaNN attribute), [1] (model.PropFormulaNN attribute), [1] ParseTree (class in model.boolean) PertinentNegativesDataset (class in datagenerator) (class in datagenerator.pertinent_negatives) PertinentNN (class in model) (class in model.pertinent_negative) perturb_func_constructor() (in module metrics) (in module metrics.metrics_wrapper) perturb_function() (datagenerator.BaseFeaturesDataset method) (datagenerator.boolean.BooleanDataset method) (datagenerator.BooleanDataset method) (datagenerator.data_generation.BaseFeaturesDataset method) perturb_standard_normal() (in module metrics) (in module metrics.metrics_wrapper) pipeline module Pipeline (class in pipeline) (class in pipeline.pipeline) pipeline.experiment module pipeline.pipeline module pipeline.results module

pn_features (datagenerator.pertinent_negatives.PertinentNegativesDataset attribute), [1] (datagenerator.PertinentNegativesDataset attribute), [1] pn_weight_factor (datagenerator.pertinent_negatives.PertinentNegativesDataset attribute), [1] (datagenerator.PertinentNegativesDataset attribute), [1] pn_zero_likelihood (datagenerator.pertinent_negatives.PertinentNegativesDataset attribute), [1] (datagenerator.PertinentNegativesDataset attribute), [1] pos_type (in module fine_tune) position (datagenerator.image_generation.ImageBuilder attribute), [1] (datagenerator.image_generation.ImageDataset attribute) (datagenerator.ImageDataset attribute) POSSIBLE_TYPE (in module common_lib) print_all_results() (pipeline.Results method) (pipeline.results.Results method) print_max_examples() (pipeline.Results method) (pipeline.results.Results method) print_min_examples() (pipeline.Results method) (pipeline.results.Results method) print_stats() (pipeline.Results method) (pipeline.results.Results method) process_data() (pipeline.Results method) (pipeline.results.Results method) PropFormulaNN (class in model) (class in model.boolean)

R

raw_data (pipeline.Results attribute) (pipeline.results.Results attribute) resize_overlay_to_background() (datagenerator.image_generation.ImageBuilder method) Results (class in pipeline) (class in pipeline.results) results (pipeline.BasePipeline attribute), [1] (pipeline.pipeline.BasePipeline attribute), [1]

rotation (datagenerator.image_generation.ImageBuilder attribute), [1] (datagenerator.image_generation.ImageDataset attribute) (datagenerator.ImageDataset attribute) run() (pipeline.ExperimentPipeline method) (pipeline.Pipeline method) (pipeline.pipeline.ExperimentPipeline method) (pipeline.pipeline.Pipeline method)

S

sample_std_dev (datagenerator.BaseFeaturesDataset attribute) (datagenerator.data_generation.BaseFeaturesDataset attribute) samples (datagenerator.BaseFeaturesDataset attribute) (datagenerator.data_generation.BaseFeaturesDataset attribute) (datagenerator.image_generation.ImageDataset attribute), [1] (datagenerator.ImageDataset attribute), [1] save() (pipeline.Results method) (pipeline.results.Results method) save_dataset() (datagenerator.BaseFeaturesDataset method) (datagenerator.data_generation.BaseFeaturesDataset method) scheduler (trainer.AutoTrainer attribute) (trainer.trainer.AutoTrainer attribute) scheduler_params (trainer.AutoTrainer attribute) (trainer.trainer.AutoTrainer attribute) score (pipeline.Example attribute) (pipeline.results.Example attribute) score_for_ranking (pipeline.Example attribute) (pipeline.results.Example attribute) seed (datagenerator.BaseFeaturesDataset attribute) (datagenerator.boolean.BooleanAndDataset attribute) (datagenerator.boolean.BooleanDataset attribute) (datagenerator.boolean.BooleanOrDataset attribute) (datagenerator.BooleanAndDataset attribute) (datagenerator.BooleanDataset attribute) (datagenerator.BooleanOrDataset attribute) (datagenerator.data_generation.BaseFeaturesDataset attribute) (datagenerator.image_generation.ImageDataset attribute), [1] (datagenerator.ImageDataset attribute), [1] (datagenerator.interacting_features.InteractingFeatureDataset attribute) (datagenerator.InteractingFeatureDataset attribute) (datagenerator.text_datasets.TextTriggerDataset attribute) (datagenerator.TextTriggerDataset attribute) SEED (in module common_lib) seeds (pipeline.Experiment attribute), [1] (pipeline.experiment.Experiment attribute), [1] set_default_target() (pipeline.BasePipeline method) (pipeline.pipeline.BasePipeline method) shape_colors (datagenerator.image_generation.ImageDataset attribute), [1] (datagenerator.ImageDataset attribute), [1] shape_id_map (datagenerator.foregrounds.DinosaurShapeGenerator attribute), [1] (datagenerator.foregrounds.GeometricShapeGenerator attribute), [1] shape_names (datagenerator.foregrounds.DinosaurShapeGenerator attribute), [1] (datagenerator.foregrounds.ForegroundGenerator attribute) (datagenerator.foregrounds.GeometricShapeGenerator attribute), [1]

shape_type (datagenerator.image_generation.ImageBuilder attribute), [1] (datagenerator.image_generation.ImageDataset attribute) (datagenerator.ImageDataset attribute) shapes (datagenerator.image_generation.ImageDataset attribute), [1] (datagenerator.ImageDataset attribute), [1] ShatteredGradientsDataset (class in datagenerator) (class in datagenerator.shattered_grad) ShatteredGradientsNN (class in model) (class in model.shattered_gradients) show_image() (datagenerator.image_generation.ImageDataset static method) (datagenerator.ImageDataset static method) shuffle_dataset() (datagenerator.image_generation.ImageDataset method) (datagenerator.ImageDataset method) shuffled (datagenerator.image_generation.ImageDataset attribute), [1] (datagenerator.ImageDataset attribute), [1] size (datagenerator.foregrounds.GeometricShapeGenerator attribute), [1] skip_tokens (datagenerator.text_datasets.TextTriggerDataset attribute) (datagenerator.TextTriggerDataset attribute) split() (datagenerator.BaseFeaturesDataset method) (datagenerator.data_generation.BaseFeaturesDataset method) subset_attribute (datagenerator.BaseFeaturesDataset attribute), [1] (datagenerator.boolean.BooleanDataset attribute) (datagenerator.BooleanDataset attribute) (datagenerator.data_generation.BaseFeaturesDataset attribute), [1] (datagenerator.data_generation.WeightedFeaturesDataset attribute) (datagenerator.interacting_features.InteractingFeatureDataset attribute), [1] (datagenerator.InteractingFeatureDataset attribute), [1] (datagenerator.pertinent_negatives.PertinentNegativesDataset attribute), [1] (datagenerator.PertinentNegativesDataset attribute), [1] (datagenerator.uncertainty_aware.UncertaintyAwareDataset attribute) (datagenerator.UncertaintyAwareDataset attribute) (datagenerator.WeightedFeaturesDataset attribute) subset_data (datagenerator.BaseFeaturesDataset attribute), [1] (datagenerator.boolean.BooleanDataset attribute) (datagenerator.BooleanDataset attribute) (datagenerator.conflicting.ConflictingDataset attribute) (datagenerator.ConflictingDataset attribute) (datagenerator.data_generation.BaseFeaturesDataset attribute), [1] (datagenerator.data_generation.WeightedFeaturesDataset attribute) (datagenerator.pertinent_negatives.PertinentNegativesDataset attribute), [1] (datagenerator.PertinentNegativesDataset attribute), [1] (datagenerator.uncertainty_aware.UncertaintyAwareDataset attribute) (datagenerator.UncertaintyAwareDataset attribute) (datagenerator.WeightedFeaturesDataset attribute) subtrees (model.boolean.ParseTree attribute)

T

target (datagenerator.text_datasets.TextTriggerDataset attribute) (datagenerator.TextTriggerDataset attribute) (pipeline.Example attribute) (pipeline.results.Example attribute) test_eval (trainer.AutoTrainer attribute), [1] (trainer.trainer.AutoTrainer attribute), [1] test_messages_len (in module fine_tune) test_step() (trainer.AutoTrainer method) (trainer.trainer.AutoTrainer method) TextTriggerDataset (class in datagenerator) (class in datagenerator.text_datasets) to_device() (in module pipeline) (in module pipeline.pipeline)

tokenizer (datagenerator.text_datasets.TextTriggerDataset attribute) (datagenerator.TextTriggerDataset attribute) train_messages_len (in module fine_tune) trainer module trainer.trainer module training_step() (trainer.AutoTrainer method) (trainer.trainer.AutoTrainer method) transform (datagenerator.image_generation.ImageDataset attribute), [1] (datagenerator.ImageDataset attribute), [1] trial_group_name (pipeline.Pipeline attribute) (pipeline.pipeline.Pipeline attribute)

U

UncertaintyAwareDataset (class in datagenerator) (class in datagenerator.uncertainty_aware) UncertaintyNN (class in model) (class in model.uncertainty_model)

unpack_batch() (pipeline.BasePipeline method) (pipeline.pipeline.BasePipeline method) upload_to_hub() (in module common_lib)

V

validate_color() (datagenerator.foregrounds.ForegroundGenerator method)

validation_step() (trainer.AutoTrainer method) (trainer.trainer.AutoTrainer method)

W

w (in module datagenerator.uncertainty_aware) (in module model.generic) (in module model.uncertainty_model) weight_range (datagenerator.data_generation.WeightedFeaturesDataset attribute) (datagenerator.interacting_features.InteractingFeatureDataset attribute) (datagenerator.InteractingFeatureDataset attribute) (datagenerator.shattered_grad.ShatteredGradientsDataset attribute) (datagenerator.ShatteredGradientsDataset attribute) (datagenerator.WeightedFeaturesDataset attribute) weighted_samples (datagenerator.data_generation.WeightedFeaturesDataset attribute), [1] (datagenerator.shattered_grad.ShatteredGradientsDataset attribute) (datagenerator.ShatteredGradientsDataset attribute) (datagenerator.uncertainty_aware.UncertaintyAwareDataset attribute), [1] (datagenerator.UncertaintyAwareDataset attribute), [1] (datagenerator.WeightedFeaturesDataset attribute), [1] WeightedFeaturesDataset (class in datagenerator) (class in datagenerator.data_generation)

weights (datagenerator.data_generation.WeightedFeaturesDataset attribute) (datagenerator.interacting_features.InteractingFeatureDataset attribute) (datagenerator.InteractingFeatureDataset attribute) (datagenerator.shattered_grad.ShatteredGradientsDataset attribute) (datagenerator.ShatteredGradientsDataset attribute) (datagenerator.uncertainty_aware.UncertaintyAwareDataset attribute), [1] (datagenerator.UncertaintyAwareDataset attribute), [1] (datagenerator.WeightedFeaturesDataset attribute) (in module model.shattered_gradients) wrap_method() (in module methods) (in module methods.methods_wrapper) wrap_method_llm() (in module methods) (in module methods.methods_wrapper) wrap_metric() (in module metrics) (in module metrics.metrics_wrapper)

Y

y_labels (pipeline.Example attribute) (pipeline.results.Example attribute)

Z

zero_likelihood (datagenerator.interacting_features.InteractingFeatureDataset attribute), [1] (datagenerator.InteractingFeatureDataset attribute), [1]