MCPcopy Create free account

hub / github.com/MLforHealth/S2SD / functions

Functions226 in github.com/MLforHealth/S2SD

↓ 40 callersMethodlog
(self, group, segments, content)
utilities/logger.py:14
↓ 4 callersMethodhistogram
Compute the histogram over similarities. Args: unique_sim_rep: torch tensor of shape nbins x n_unique_neg_similariti
criteria/histogram.py:81
↓ 4 callersMethodupdate
(self, *sub_loggers, all=False)
utilities/logger.py:134
↓ 3 callersMethodget_distmat
(self, arr)
datasampler/disthist_batchmatch_sampler.py:99
↓ 3 callersMethodsmat
(self, A, B, mode='cosine')
criteria/s2sd.py:140
↓ 2 callersMethodgreedy_coreset
(self, A, samples)
datasampler/samplers.py:108
↓ 2 callersMethodinverse_sphere_distances
(self, batch, anchor_to_all_dists, labels, anchor_label)
batchminer/diva_shared_distance_apn.py:37
↓ 2 callersMethodkl_div
(self, A, B)
criteria/s2sd.py:132
↓ 2 callersMethodpdistsq
(self, A)
datasampler/samplers.py:103
↓ 2 callersFunctionvisualize_scatter
(features, labels, perplex, figsize=(10,10), save_path=None)
utilities/visualize_tsne.py:93
↓ 1 callersMethodbackward
Container to reverse gradient signal during backward pass. Input: grad_output: any computed gradient.
criteria/adversarial_separation.py:75
↓ 1 callersMethodcompute_standard
(self, opt, model, dataloader, evaltypes, device, **kwargs)
metrics/__init__.py:69
↓ 1 callersMethodcreate_storage
(self, dataloader, model, device)
datasampler/samplers.py:34
↓ 1 callersMethodd2_coreset
datasampler/d2_coreset_sampler.py:90
↓ 1 callersFunctiondistance_measure
(embeddings, labels)
evaluation/evaluate.py:38
↓ 1 callersMethodensure_3dim
(self, img)
datasets/basic_dataset_scaffold.py:72
↓ 1 callersFunctionevaluate_one_dataset
(LOG, dataloader, model, opt, spliteval=True, evaltypes=['Class'], save=True, give_return=False, aux_store=Non
evaluation/evaluate.py:62
↓ 1 callersMethodfull_storage_update
(self, dataloader, model, device)
datasampler/greedy_coreset_sampler.py:93
↓ 1 callersFunctiongimme_save_string
(opt)
utilities/misc.py:16
↓ 1 callersFunctiongrad_reverse
Applies gradient reversal on input. Input: x: any torch tensor input.
criteria/adversarial_separation.py:85
↓ 1 callersMethodgreedy_coreset
datasampler/greedy_coreset_sampler.py:111
↓ 1 callersMethodinit_setup
(self)
datasets/basic_dataset_scaffold.py:50
↓ 1 callersMethodinverse_sphere_distances
(self, batch, anchor_to_all_dists, labels, anchor_label)
batchminer/random_distance.py:38
↓ 1 callersMethodinverse_sphere_distances
(self, batch, anchor_to_all_dists, labels, anchor_label)
batchminer/easypositive.py:36
↓ 1 callersMethodinverse_sphere_distances
(self, batch, anchor_to_all_dists, labels, anchor_label)
batchminer/epdn.py:36
↓ 1 callersMethodinverse_sphere_distances
(self, dim, bs, anchor_to_all_dists, labels, anchor_label)
batchminer/distance.py:48
↓ 1 callersMethodinverse_sphere_distances
(self, batch, anchor_to_all_dists, labels, anchor_label)
batchminer/rho_distance.py:50
↓ 1 callersMethodinverse_sphere_distances
(self, batch, anchor_to_all_dists, labels, anchor_label)
batchminer/diva_shared_distance_an.py:36
↓ 1 callersMethodmake_plot
(self, base_title, title_append, sub_plots, sub_plots_data)
utilities/logger.py:37
↓ 1 callersMethodpdist
(self, A)
batchminer/diva_shared_distance_apn.py:56
↓ 1 callersMethodpdist
(self, A)
batchminer/epshn.py:41
↓ 1 callersMethodpdist
(self, A)
batchminer/random_distance.py:57
↓ 1 callersMethodpdist
(self, A)
batchminer/ephn.py:44
↓ 1 callersMethodpdist
(self, A)
batchminer/easypositive.py:55
↓ 1 callersMethodpdist
(self, A, eps=1e-4)
batchminer/parametric.py:58
↓ 1 callersMethodpdist
(self, A)
batchminer/epdn.py:55
↓ 1 callersMethodpdist
(self, A)
batchminer/distance.py:66
↓ 1 callersMethodpdist
(self, A)
batchminer/softhard.py:50
↓ 1 callersMethodpdist
(self, A, eps=1e-4)
batchminer/rho_distance.py:69
↓ 1 callersMethodpdist
(self, A)
batchminer/semihard.py:43
↓ 1 callersMethodpdist
(self, A)
batchminer/diva_shared_distance_an.py:55
↓ 1 callersMethodprecompute_indices
(self)
datasampler/d2_coreset_sampler.py:43
↓ 1 callersMethodprep
(self, thing)
criteria/s2sd.py:76
↓ 1 callersMethodpresample_infobatch
(self, classes, A, samples)
datasampler/samplers.py:127
↓ 1 callersMethodquadruplet_distance
(self, anchor, positive, negative, fourth_negative)
criteria/quadruplet.py:30
↓ 1 callersFunctionrecover_closest_standard
(feature_matrix_all, image_paths, save_path, n_image_samples=10, n_closest=3)
evaluation/__init__.py:82
↓ 1 callersMethodreplace_storage_entries
(self, embeddings, indices)
datasampler/d2_coreset_sampler.py:72
↓ 1 callersFunctionselect
(metricname, opt)
metrics/__init__.py:12
↓ 1 callersFunctionset_checkpoint
(model, opt, progress_saver, savepath, aux=None)
evaluation/__init__.py:68
↓ 1 callersFunctionset_logging
(opt)
utilities/logger.py:64
↓ 1 callersMethodset_sample_distr
(self)
batchminer/parametric.py:65
↓ 1 callersMethodspc_batchfinder
(self, n_samples)
datasampler/fid_batchmatch_sampler.py:85
↓ 1 callersMethodspc_batchfinder
(self, n_samples)
datasampler/disthist_batchmatch_sampler.py:87
↓ 1 callersFunctiontSNE_test_train
(opt, dataloaders, model, n_samples=1000, perplex=40.0)
utilities/visualize_tsne.py:53
↓ 1 callersMethodtriplet_distance
(self, anchor, positive, negative)
criteria/triplet.py:24
↓ 1 callersMethodtriplet_distance
(self, anchor, positive, negative)
criteria/quadruplet.py:27
FunctionGive
(opt, datapath)
datasets/inshop.py:5
FunctionGive
(opt, datapath)
datasets/stanford_online_products.py:6
FunctionGive
(opt, datapath)
datasets/cub200.py:4
FunctionGive
(opt, datapath)
datasets/cars196.py:5
Method__call__
(self, target_labels, features_cosine)
metrics/c_mAP_1000.py:12
Method__call__
(self, target_labels, computed_cluster_labels, features, centroids)
metrics/f1.py:10
Method__call__
(self, target_labels, k_closest_classes, **kwargs)
metrics/e_recall.py:9
Method__call__
(self, target_labels, features_cosine)
metrics/c_mAP_c.py:12
Method__call__
(self, target_labels, features)
metrics/mAP_1000.py:12
Method__call__
(self, target_labels, features)
metrics/mAP_c.py:12
Method__call__
(self, target_labels, features)
metrics/mAP_lim.py:12
Method__call__
(self, features)
metrics/rho_spectrum.py:13
Method__call__
(self, target_labels, computed_cluster_labels_cosine, features_cosine, centroids_cosine)
metrics/c_f1.py:10
Method__call__
(self, target_labels, features_cosine)
metrics/c_mAP_lim.py:12
Method__call__
(self, target_labels, features)
metrics/mAP.py:12
Method__call__
(self, target_labels, k_closest_classes_cosine, **kwargs)
metrics/c_recall.py:9
Method__call__
(self, target_labels, computed_cluster_labels_cosine)
metrics/c_nmi.py:8
Method__call__
(self, target_labels, computed_cluster_labels)
metrics/nmi.py:8
Method__call__
(self, features, target_labels)
metrics/dists.py:12
Method__call__
(self, batch, labels)
batchminer/diva_shared_distance_apn.py:11
Method__call__
(self, batch, labels, return_distances=False)
batchminer/epshn.py:9
Method__call__
(self, batch, labels)
batchminer/random_distance.py:11
Method__call__
(self, batch, labels)
batchminer/npair.py:7
Method__call__
(self, batch, labels, return_distances=False)
batchminer/ephn.py:9
Method__call__
(self, batch, labels)
batchminer/random.py:10
Method__call__
(self, batch, labels)
batchminer/easypositive.py:11
Method__call__
(self, batch, labels)
batchminer/parametric.py:17
Method__call__
(self, batch, labels)
batchminer/epdn.py:11
Method__call__
(self, batch, labels)
batchminer/lifted.py:8
Method__call__
(self, batch, labels, tar_labels=None, return_distances=False, distances=None)
batchminer/distance.py:13
Method__call__
(self, batch, labels)
batchminer/intra_random.py:10
Method__call__
(self, batch, labels, return_distances=False)
batchminer/softhard.py:9
Method__call__
(self, batch, labels, return_distances=False)
batchminer/rho_distance.py:13
Method__call__
(self, batch, labels, return_distances=False)
batchminer/semihard.py:10
Method__call__
(self, batch, labels)
batchminer/diva_shared_distance_an.py:11
Method__getitem__
(self, idx)
datasets/basic_dataset_scaffold.py:78
Method__init__
(self, **kwargs)
metrics/c_mAP_1000.py:8
Method__init__
(self, **kwargs)
metrics/f1.py:6
Method__init__
(self, k, **kwargs)
metrics/e_recall.py:4
Method__init__
(self, **kwargs)
metrics/c_mAP_c.py:8
Method__init__
(self, **kwargs)
metrics/mAP_1000.py:8
Method__init__
(self, k, **kwargs)
metrics/a_recall.py:4
Method__init__
(self, **kwargs)
metrics/mAP_c.py:8
Method__init__
(self, **kwargs)
metrics/mAP_lim.py:8
next →1–100 of 226, ranked by callers