MCPcopy Create free account

hub / github.com/YyzHarry/imbalanced-regression / functions

Functions270 in github.com/YyzHarry/imbalanced-regression

⨍ Functions 270◇ Types & classes 47
↓ 7 callersMethodupdate
(self, val, n=1)
agedb-dir/utils.py:22
↓ 7 callersMethodupdate
(self, val, n=1)
imdb-wiki-dir/utils.py:22
↓ 6 callersFunctiondevice_mapping
In order to `torch.load()` a GPU-trained model onto a CPU (or specific GPU), you have to supply a `map_location` function. Call this with
sts-b-dir/util.py:35
↓ 4 callersMethod__init__
(self, num_features = 2048)
nyud2-dir/models/modules.py:63
↓ 4 callersFunction_is_pil_image
(img)
nyud2-dir/nyu_transform.py:13
↓ 4 callersMethod_make_layer
(self, block, planes, blocks, stride=1)
agedb-dir/resnet.py:111
↓ 4 callersMethod_make_layer
(self, block, planes, blocks, stride=1)
imdb-wiki-dir/resnet.py:111
↓ 4 callersMethod_make_layer
(self, block, planes, blocks, stride=1)
nyud2-dir/models/resnet.py:119
↓ 4 callersMethodget_metrics
(self, reset=False, type=None)
sts-b-dir/tasks.py:103
↓ 4 callersMethodtrain
(self, tasks, validation_interval, train_params, optimizer_params, resume=False)
sts-b-dir/trainer.py:99
↓ 3 callersMethod_forward
(self, batch, epoch=None, task=None)
sts-b-dir/trainer.py:329
↓ 3 callersFunctioncalibrate_mean_var
(matrix, m1, v1, m2, v2, clip_min=0.1, clip_max=10)
agedb-dir/utils.py:97
↓ 3 callersFunctioncalibrate_mean_var
(matrix, m1, v1, m2, v2, clip_min=0.1, clip_max=10)
imdb-wiki-dir/utils.py:97
↓ 3 callersMethodfrom_params
Generator trainer from parameters.
sts-b-dir/trainer.py:408
↓ 3 callersFunctionload_tsv
Load a tsv
sts-b-dir/tasks.py:13
↓ 3 callersFunctionprocess_split
Convert a dataset of sentences into padded sequences of indices.
sts-b-dir/preprocess.py:150
↓ 3 callersFunctionsave_checkpoint
(state_dict, epoch, error, metric_dict, filename='checkpoint.pth.tar')
nyud2-dir/train.py:254
↓ 3 callersMethodto_tensor
(self, pic)
nyud2-dir/nyu_transform.py:175
↓ 3 callersFunctionvalidate
(val_loader, model, train_labels=None, prefix='Val')
agedb-dir/train.py:286
↓ 3 callersFunctionvalidate
(val_loader, model, train_labels=None, prefix='Val')
imdb-wiki-dir/train.py:286
↓ 2 callersMethod__init__
(self, block, layers, fds, bucket_num, bucket_start, start_update, start_smooth, kernel, ks,
agedb-dir/resnet.py:75
↓ 2 callersMethod__init__
(self, block, layers, fds, bucket_num, bucket_start, start_update, start_smooth, kernel, ks,
imdb-wiki-dir/resnet.py:75
↓ 2 callersMethod__init__
(self, block, layers, num_classes=1000)
nyud2-dir/models/resnet.py:96
↓ 2 callersMethod_get_bucket_idx
(self, label)
sts-b-dir/fds.py:51
↓ 2 callersMethod_running_stats_to_device
(self, device)
nyud2-dir/models/fds.py:88
↓ 2 callersMethod_save_checkpoint
(self, training_state, best=False)
sts-b-dir/trainer.py:336
↓ 2 callersMethodcenterCrop
(self, image, size)
nyud2-dir/nyu_transform.py:134
↓ 2 callersMethodchangeScale
(self, img, size, interpolation=Image.BILINEAR)
nyud2-dir/nyu_transform.py:94
↓ 2 callersFunctionconv3x3
3x3 convolution with padding
agedb-dir/resnet.py:9
↓ 2 callersFunctionconv3x3
3x3 convolution with padding
imdb-wiki-dir/resnet.py:9
↓ 2 callersFunctionconv3x3
3x3 convolution with padding
nyud2-dir/models/resnet.py:17
↓ 2 callersFunctioncount_sentence
Update counts for words in the sentence
sts-b-dir/preprocess.py:81
↓ 2 callersFunctioncreate
(db)
imdb-wiki-dir/data/create_imdb_wiki.py:56
↓ 2 callersMethoddisplay
(self, batch)
agedb-dir/utils.py:39
↓ 2 callersMethoddisplay
(self, batch)
imdb-wiki-dir/utils.py:39
↓ 2 callersMethodevaluate
(output, target)
nyud2-dir/util.py:89
↓ 2 callersMethodforward
(self, task=None, epoch=None, input1=None, input2=None, mask1=None, mask2=None, label=None, weight=None)
sts-b-dir/models.py:84
↓ 2 callersFunctionget_args
()
imdb-wiki-dir/data/create_imdb_wiki.py:48
↓ 2 callersFunctionget_bin_idx
(label)
sts-b-dir/tasks.py:51
↓ 2 callersFunctionlg10
(x)
nyud2-dir/util.py:8
↓ 2 callersFunctionload_data
(args)
nyud2-dir/preprocess_nyud2.py:10
↓ 2 callersFunctionprocess_sentence
process a sentence using NLTK toolkit
sts-b-dir/tasks.py:9
↓ 2 callersFunctionresume_checkpoint
(model, model_state, backbone_only=False)
sts-b-dir/util.py:75
↓ 2 callersFunctiontest
(test_loader, model)
nyud2-dir/test.py:39
↓ 2 callersMethodupdate
(self, val, n=1)
nyud2-dir/train.py:247
↓ 1 callersMethod__init__
(self, args, pair_encoder, FDS=None)
sts-b-dir/models.py:72
↓ 1 callersMethod_aggregate_results
Ad hoc helper function to print results after finishing training
sts-b-dir/trainer.py:240
↓ 1 callersMethod_check_history
Given a task, the history of the performance on that task, and the current score, check if current score is best so far and i
sts-b-dir/trainer.py:50
↓ 1 callersMethod_check_stop
Check to see if should stop
sts-b-dir/trainer.py:315
↓ 1 callersMethod_description_from_metrics
(self, metrics)
sts-b-dir/trainer.py:333
↓ 1 callersMethod_get_batch_fmtstr
(num_batches)
agedb-dir/utils.py:45
↓ 1 callersMethod_get_batch_fmtstr
(num_batches)
imdb-wiki-dir/utils.py:45
↓ 1 callersFunction_get_bin_idx
(label)
sts-b-dir/glue_data/create_sts.py:44
↓ 1 callersMethod_get_bucket_idx
(self, label)
nyud2-dir/models/fds.py:51
↓ 1 callersMethod_get_bucket_weights
(self, args)
nyud2-dir/loaddata.py:29
↓ 1 callersMethod_get_kernel_window
(kernel, ks, sigma)
agedb-dir/fds.py:38
↓ 1 callersMethod_get_kernel_window
(kernel, ks, sigma)
sts-b-dir/fds.py:34
↓ 1 callersMethod_get_kernel_window
(kernel, ks, sigma)
imdb-wiki-dir/fds.py:38
↓ 1 callersMethod_get_kernel_window
(kernel, ks, sigma)
nyud2-dir/models/fds.py:35
↓ 1 callersMethod_get_weights
(self, depth)
nyud2-dir/loaddata.py:58
↓ 1 callersFunction_is_numpy_image
(img)
nyud2-dir/nyu_transform.py:20
↓ 1 callersMethod_prepare_weights
(self, reweight, max_target=121, lds=False, lds_kernel='gaussian', lds_ks=5, lds_sigma=2)
agedb-dir/datasets.py:55
↓ 1 callersMethod_prepare_weights
(self, reweight, max_target=121, lds=False, lds_kernel='gaussian', lds_ks=5, lds_sigma=2)
imdb-wiki-dir/datasets.py:55
↓ 1 callersMethod_restore_checkpoint
Restores a model from a serialization_dir to the last saved checkpoint. This includes an epoch count and optimizer state, which is se
sts-b-dir/trainer.py:367
↓ 1 callersMethod_setup_training
(self, tasks, train_params, optimizer_params, iterator)
sts-b-dir/trainer.py:77
↓ 1 callersMethod_update_last_epoch_stats
(self)
agedb-dir/fds.py:54
↓ 1 callersMethod_update_last_epoch_stats
(self)
sts-b-dir/fds.py:59
↓ 1 callersMethod_update_last_epoch_stats
(self)
imdb-wiki-dir/fds.py:54
↓ 1 callersMethod_update_last_epoch_stats
(self)
nyud2-dir/models/fds.py:55
↓ 1 callersMethod_validate
(self, epoch, val_check, tasks, task_infos, metric_infos, iterator)
sts-b-dir/trainer.py:263
↓ 1 callersFunctionadjust_learning_rate
(optimizer, epoch, args)
agedb-dir/utils.py:81
↓ 1 callersFunctionadjust_learning_rate
(optimizer, epoch, args)
imdb-wiki-dir/utils.py:81
↓ 1 callersFunctionadjust_learning_rate
(optimizer, epoch)
nyud2-dir/train.py:230
↓ 1 callersFunctionbuild_model
Build model according to arguments
sts-b-dir/models.py:16
↓ 1 callersFunctionbuild_regressor
Build the regressor
sts-b-dir/models.py:62
↓ 1 callersMethodbuild_regressor
(self, task, d_inp)
sts-b-dir/models.py:80
↓ 1 callersFunctionbuild_tasks
Prepare tasks
sts-b-dir/preprocess.py:24
↓ 1 callersFunctionbuild_trainer
Build a trainer
sts-b-dir/trainer.py:19
↓ 1 callersFunctioncalc_age
(taken, dob)
imdb-wiki-dir/data/create_imdb_wiki.py:10
↓ 1 callersFunctioncalibrate_mean_var
(matrix, m1, v1, m2, v2, clip_min=0.5, clip_max=2.)
sts-b-dir/util.py:63
↓ 1 callersFunctioncalibrate_mean_var
(matrix, m1, v1, m2, v2, clip_min=0.2, clip_max=5.)
nyud2-dir/util.py:151
↓ 1 callersFunctioncombine_dataset
(path='meta')
imdb-wiki-dir/data/create_imdb_wiki.py:37
↓ 1 callersFunctioncreate_FDS_train_subset
(args)
nyud2-dir/preprocess_nyud2.py:44
↓ 1 callersFunctioncreate_FDS_train_subset_id
(args)
nyud2-dir/preprocess_nyud2.py:34
↓ 1 callersFunctioncreate_balanced_testset
(args)
nyud2-dir/preprocess_nyud2.py:53
↓ 1 callersFunctiondefine_model
(args)
nyud2-dir/train.py:59
↓ 1 callersFunctiondefine_model
()
nyud2-dir/test.py:61
↓ 1 callersFunctiondel_field_tokens
Save memory by deleting the tokens that will no longer be used
sts-b-dir/preprocess.py:51
↓ 1 callersFunctionevaluate
Evaluate on a dataset
sts-b-dir/evaluate.py:5
↓ 1 callersMethodevaluate_shot
(self)
nyud2-dir/util.py:53
↓ 1 callersFunctiongetNanMask
(x)
nyud2-dir/util.py:20
↓ 1 callersFunctionget_args
()
agedb-dir/data/create_agedb.py:7
↓ 1 callersFunctionget_bin_idx
(x)
nyud2-dir/preprocess_nyud2.py:50
↓ 1 callersMethodget_bin_idx
(self, x)
nyud2-dir/loaddata.py:55
↓ 1 callersMethodget_bin_idx
(x)
nyud2-dir/util.py:85
↓ 1 callersFunctionget_embeddings
Get embeddings for the words in vocab
sts-b-dir/preprocess.py:110
↓ 1 callersFunctionget_lds_kernel_window
(kernel, ks, sigma)
agedb-dir/utils.py:110
↓ 1 callersFunctionget_lds_kernel_window
(kernel, ks, sigma)
sts-b-dir/util.py:86
↓ 1 callersFunctionget_lds_kernel_window
(kernel, ks, sigma)
imdb-wiki-dir/utils.py:110
↓ 1 callersFunctionget_lds_kernel_window
(kernel, ks, sigma)
nyud2-dir/util.py:164
next →1–100 of 270, ranked by callers