MCPcopy Create free account

hub / github.com/HEmile/neurosymbolic-diffusion / functions

Functions1,084 in github.com/HEmile/neurosymbolic-diffusion

↓ 91 callersMethodto
(self, device)
expressive/experiments/rsbench/models/xordpl.py:235
↓ 75 callersMethodto
(self, device)
expressive/experiments/rsbench/utils/tcav/tcav/model_wrapper.py:65
↓ 38 callersFunctionadd_management_args
Adds the arguments used in management Args: parser: the parser instance Returns: None: This function does not return a value
expressive/experiments/rsbench/utils/args.py:188
↓ 37 callersFunctionadd_experiment_args
Adds the arguments used by all the models. Args: parser: the parser instance Returns: None: This function does not return a
expressive/experiments/rsbench/utils/args.py:8
↓ 37 callersFunctionget_device
(args)
expressive/experiments/rsbench/utils/conf.py:9
↓ 33 callersFunctionget_loader
(dataset, batch_size, num_workers=4, val_test=False, sampler=None)
expressive/experiments/rsbench/datasets/utils/base_dataset.py:77
↓ 30 callersFunctiontest
(a, b, name)
expressive/experiments/stats.py:5
↓ 19 callersMethodnorm
(x)
expressive/methods/logger.py:105
↓ 17 callersMethodeval
(self)
expressive/experiments/rsbench/utils/tcav/tcav/model_wrapper.py:59
↓ 14 callersMethodget_data_loaders
(self)
expressive/experiments/rsbench/datasets/boia.py:27
↓ 12 callersFunctionSDDOIA_get_loader
(dataset, batch_size, num_workers=4, val_test=False)
expressive/experiments/rsbench/datasets/utils/base_dataset.py:134
↓ 12 callersMethodstep
(self)
expressive/methods/logger.py:212
↓ 11 callersFunctioneval
( val_loader: DataLoader, test_logger: TestLog, model: BaseNeSyDiffusion, device: torch.device
expressive/experiments/rsbench/nesydiffusion.py:61
↓ 11 callersMethody_from_w
(self, w_SBW: Tensor)
expressive/methods/base_model.py:26
↓ 10 callersMethod__init__
(self, dim: int)
expressive/models/diffusion_model.py:290
↓ 10 callersFunctionhidden_size
(model: str)
expressive/models/dit.py:353
↓ 10 callersMethodmask_dim_w
(self)
expressive/methods/base_model.py:510
↓ 10 callersFunctionouter_product
(*tensors)
expressive/experiments/rsbench/models/utils/ops.py:4
↓ 9 callersFunctionBOIA_get_loader
(dataset, batch_size, val_test)
expressive/experiments/rsbench/datasets/utils/base_dataset.py:152
↓ 9 callersMethodloss_weight
(self, losses: Tensor, time: Tensor)
expressive/methods/base_model.py:390
↓ 9 callersMethodmask_dim_y
(self)
expressive/methods/base_model.py:513
↓ 9 callersMethodnormalize
(self, a, z)
expressive/experiments/rsbench/models/utils/deepproblog_modules.py:56
↓ 9 callersFunctionsafe_sample_categorical
( distr: torch.distributions.Categorical, shape: Optional[torch.Size] = None )
expressive/util.py:24
↓ 8 callersFunctionbuild_worlds_queries_matrix
Build Worlds-Queries matrix
expressive/experiments/rsbench/models/utils/utils_problog.py:330
↓ 8 callersFunctionfprint
Flushing print Args: args: arguments kwargs: key-value arguments Returns: None: This function does not return a valu
expressive/experiments/rsbench/utils/__init__.py:20
↓ 8 callersMethodshape_w
(self)
expressive/methods/base_model.py:18
↓ 7 callersMethodto
(self, device)
expressive/experiments/rsbench/models/sddoiadpl.py:351
↓ 6 callersMethod__init__
( self, embed_dim: int, # vision image_resolution: int, vision_layers:
expressive/experiments/rsbench/preprocessing/clip/model.py:311
↓ 6 callersFunctioncompute_coverage
Compute the coverage of a confusion matrix. Essentially this metric is
expressive/experiments/rsbench/utils/train.py:46
↓ 6 callersMethoddistribution
( self, x_t: DATA, encoding: Tensor, t: TimeSteps = None, s: TimeSteps
expressive/models/diffusion_model.py:235
↓ 6 callersFunctionload
Load a CLIP model Parameters ---------- name : str A model name listed by `clip.available_models()`, or the path to a model check
expressive/experiments/rsbench/preprocessing/clip/clip.py:116
↓ 6 callersMethodsample
Initialize common components for sampling methods. Args: x_BX: Input tensor w_T_BW: Initial w tensor (can be masked)
expressive/methods/base_model.py:185
↓ 5 callersMethod_create_dir
(self, directory)
expressive/experiments/rsbench/datasets/clipshortcutmnist.py:393
↓ 5 callersMethod_matvecmul
(self, Q_TKK: Tensor, x_BTNK: Tensor)
expressive/models/diffusion_model.py:102
↓ 5 callersMethodevaluate
( self, x_BX: torch.Tensor, y_0_BY: torch.Tensor, w_0_BW: Optional[torch.Tenso
expressive/methods/base_model.py:394
↓ 5 callersFunctiongenerate_r_seq
(size)
expressive/experiments/rsbench/datasets/utils/mnist_creation.py:245
↓ 5 callersFunctionplot_confusion_matrix
Generate and plot a confusion matrix using Matplotlib with normalization. Parameters: y_true (array-like): Ground truth labels.
expressive/experiments/rsbench/utils/train.py:61
↓ 5 callersMethodshape_y
(self)
expressive/methods/base_model.py:22
↓ 5 callersMethodt_step
(self, x_0: DATA, t: TimeSteps = None)
expressive/models/diffusion_model.py:29
↓ 5 callersMethodtrain
(self, activations)
expressive/experiments/rsbench/utils/tcav/tcav/cav.py:36
↓ 4 callersFunctionMNMATH_get_loader
(dataset, batch_size, val_test)
expressive/experiments/rsbench/datasets/utils/base_dataset.py:190
↓ 4 callersFunctionXOR_get_loader
(dataset, batch_size, val_test)
expressive/experiments/rsbench/datasets/utils/base_dataset.py:178
↓ 4 callersMethod__init__
(self, num_classes=2)
expressive/experiments/rsbench/backbones/cnnnosharing.py:54
↓ 4 callersMethod__init__
Initialize method Args: self: instance loss: loss function nr_classes: number of classes Returns
expressive/experiments/rsbench/utils/dpl_loss.py:127
↓ 4 callersMethod_getQ_lineQ
(self, t: TimeSteps)
expressive/models/diffusion_model.py:107
↓ 4 callersFunction_get_tag
Get tag for the model name Args: args: command line arguments Returns: tag (str): tag for the model name
expressive/experiments/rsbench/utils/checkpoint.py:7
↓ 4 callersMethod_make_layer
(self, planes, blocks, stride=1)
expressive/experiments/rsbench/preprocessing/clip/model.py:159
↓ 4 callersFunctionconvert_to_categories
(elements)
expressive/experiments/rsbench/utils/train.py:31
↓ 4 callersFunctioncreate_path
Create path function, create folder if it does not exists Args: path (str): path value Returns: None: This function does not
expressive/experiments/rsbench/utils/conf.py:58
↓ 4 callersFunctiondata_loader
(base_path, dataset_name)
expressive/experiments/rsbench/utils/tcav/tcav/main.py:36
↓ 4 callersMethodencode_image
(self, image)
expressive/experiments/rsbench/preprocessing/clip/model.py:417
↓ 4 callersFunctionevaluate_metrics
Evaluate metrics of the frequentist model Args: model (nn.Module): network loader: dataloader args: command line argument
expressive/experiments/rsbench/utils/metrics.py:90
↓ 4 callersMethodget_backbone
(self)
expressive/experiments/rsbench/datasets/boia.py:87
↓ 4 callersMethodget_concept_labels
(self)
expressive/experiments/rsbench/datasets/boia.py:96
↓ 4 callersFunctionget_dataset
Creates and returns a continual dataset. :param args: the arguments which contains the hyperparameters :return: the continual dataset
expressive/experiments/rsbench/datasets/__init__.py:29
↓ 4 callersMethodget_layer_representation
Forward method Args: self: instance x (torch.tensor): input vector Returns: out_dict: output dic
expressive/experiments/rsbench/models/mnistcbm.py:121
↓ 4 callersMethodget_w_dim
(self)
expressive/experiments/rsbench/datasets/boia.py:106
↓ 4 callersFunctionload_2MNIST
( n_digits=10, dataset_dimensions={"train": 42000, "val": 12000, "test": 6000}, c_sup=1, which
expressive/experiments/rsbench/datasets/utils/mnist_creation.py:220
↓ 4 callersMethodproblog_inference
Performs ProbLog inference to retrieve the worlds probability distribution P(w). Works with an arbitrary number of encoded bits (digits).
expressive/experiments/rsbench/models/xordpl.py:106
↓ 4 callersFunctionprogress_bar
Prints out the progress bar on the stderr file. Args: i: the current iteration max_iter: the maximum number of iteration
expressive/experiments/rsbench/utils/status.py:62
↓ 3 callersMethod__init__
(self)
expressive/experiments/rsbench/utils/tcav/tcav/l_models.py:56
↓ 3 callersMethod_create_dir
(self, directory)
expressive/experiments/rsbench/datasets/shortcutmnist.py:662
↓ 3 callersFunction_make_save_dir
creates save directory if one does not exist save_name: full save path
expressive/experiments/rsbench/preprocessing/utils.py:339
↓ 3 callersFunctionbuild_worlds_queries_matrix_KAND
Build Worlds-Queries matrix
expressive/experiments/rsbench/models/utils/utils_problog.py:387
↓ 3 callersMethodcompute_entropy
Compute the concept distribution entropy conditioned on the query.
expressive/experiments/rsbench/models/sddoiadpl.py:274
↓ 3 callersMethodcond_jump
( self, x_0: DATA, x_t: DATA, t: TimeSteps = None, s: TimeSteps = None )
expressive/models/diffusion_model.py:33
↓ 3 callersFunctioncreate_dataset
( n_digit=2, sequence_len=2, samples_x_world=100, train=True, download=False )
expressive/experiments/rsbench/preprocessing/mnist/mnist_creation.py:150
↓ 3 callersFunctioncreate_dataset
( n_digit=2, sequence_len=2, samples_x_world=100, train=True, download=False )
expressive/experiments/rsbench/datasets/utils/mnist_creation.py:129
↓ 3 callersFunctioncreate_dataset
( n_digit=2, sequence_len=2, samples_x_world=100, train=True, download=False )
expressive/experiments/rsbench/datasets/utils/clip_mnst_creation.py:269
↓ 3 callersMethodcreate_dict
(self, iterations: int)
expressive/methods/logger.py:21
↓ 3 callersFunctiondecode_label
(labels_BY, args: RSBenchArguments)
expressive/experiments/rsbench/nesydiffusion.py:47
↓ 3 callersMethodencode_x
(self, x: Tensor)
expressive/models/diffusion_model.py:381
↓ 3 callersMethodentropy_loss
(self, y_0_BY: Tensor, q_w_0_BWD: Tensor)
expressive/methods/base_model.py:345
↓ 3 callersFunctioneval
( loader: DataLoader, logger: TestLog, model: PathAbsorbing, device: torch.device, args: P
expressive/experiments/path_planning/path_planning.py:28
↓ 3 callersMethodeval_y
(self, y_0_SBY: Tensor, y_0_BY: Tensor, w_0_BW: Tensor)
expressive/methods/base_model.py:29
↓ 3 callersFunctionexpected_calibration_error
Computes the ECE Args: confs (ndarray): confidence preds (ndarray): predictions labels (ndarray): labels num_bins
expressive/experiments/rsbench/utils/metrics.py:1150
↓ 3 callersFunctionget_all_models
()
expressive/experiments/rsbench/models/__init__.py:5
↓ 3 callersMethodget_labels
(self)
expressive/experiments/rsbench/datasets/xor.py:75
↓ 3 callersMethodget_split
(self)
expressive/experiments/rsbench/datasets/boia.py:93
↓ 3 callersMethodloss
( self, x_BX: Tensor, y_0_BY: Tensor, log: TrainingLog, w_0_BW: Option
expressive/methods/base_model.py:373
↓ 3 callersFunctionmarginal_mode
(x_SBD: Tensor, dim: int=0)
expressive/util.py:37
↓ 3 callersFunctionobstacle
(c, p, r, o)
expressive/experiments/rsbench/utils/boia_ltn_loss.py:101
↓ 3 callersFunctionplot_multilabel_confusion_matrix
( y_true, y_pred, class_names, title, save_path=None )
expressive/experiments/rsbench/utils/train.py:104
↓ 3 callersMethodpush
(self, num_batches: int, extra_stats: dict={})
expressive/methods/logger.py:239
↓ 3 callersFunctionsetup
()
expressive/experiments/rsbench/utils/tcav/tcav/main.py:155
↓ 3 callersMethodto
(self, device)
expressive/experiments/rsbench/models/mnmathdpl.py:304
↓ 3 callersMethodto
(self, device)
expressive/experiments/rsbench/models/minikanddpl.py:304
↓ 3 callersFunctionvector_to_base10
(w: torch.Tensor, N: int)
expressive/experiments/mnist_op/absorbing_mnist.py:16
↓ 2 callersFunctionADDMNIST_Classification
Addmnist classification loss Args: out_dict: output dictionary args: command line arguments Returns: loss: loss valu
expressive/experiments/rsbench/utils/losses.py:7
↓ 2 callersFunctionADDMNIST_REC_Match
Addmnist concept match loss Args: out_dict: output dictionary args: command line arguments Returns: loss: loss value
expressive/experiments/rsbench/utils/losses.py:108
↓ 2 callersMethodKANDsat_agg_loss
(self, shapes, colors, conc_preds, labels, args, b_idx)
expressive/experiments/rsbench/utils/kand_ltn_loss.py:32
↓ 2 callersMethod__init__
(self)
expressive/experiments/rsbench/backbones/mnistcnn.py:7
↓ 2 callersMethod__init__
(self, x_shape=(3, 64, 64), z_size=6, z_multiplier=1)
expressive/experiments/rsbench/backbones/disent_encoder_decoder.py:17
↓ 2 callersMethod__init__
( self, seq_length: int, hidden_size: int = 1152, depth: int = 28, num
expressive/models/dit.py:121
↓ 2 callersMethod_create_dir
(self, directory)
expressive/experiments/rsbench/datasets/clipboia.py:99
↓ 2 callersMethod_create_dir
(self, directory)
expressive/experiments/rsbench/datasets/clipsddoia.py:97
↓ 2 callersFunction_transform
(n_px)
expressive/experiments/rsbench/preprocessing/clip/clip.py:96
↓ 2 callersFunctionbytes_to_unicode
Returns list of utf-8 byte and a corresponding list of unicode strings. The reversible bpe codes work on unicode strings. This means you
expressive/experiments/rsbench/preprocessing/clip/simple_tokenizer.py:18
↓ 2 callersFunctioncan_turn
(lane, green_light, follow)
expressive/experiments/rsbench/utils/boia_ltn_loss.py:183
↓ 2 callersFunctioncannot_turn
(no_lane, obstacle, solid_line)
expressive/experiments/rsbench/utils/boia_ltn_loss.py:187
next →1–100 of 1,084, ranked by callers