MCPcopy Create free account

hub / github.com/MarcCoru/locationencoder / functions

Functions175 in github.com/MarcCoru/locationencoder

Method__init__
(self, positional_encoding_name, neural_network_name, hparams)
locationencoder/locationencoder.py:73
Method__init__
(self, num_classes, lr=0.0045, momentum=0.9, weight_decay=1e-4)
locationencoder/imageencoder.py:8
Method__init__
(self, imageencoder_checkpoint, locationencoder_checkpoint, use_logits=False, lr=0.0045,
locationencoder/locationimageencoder.py:10
Method__init__
(self, linear_size)
locationencoder/nn/fcnet.py:8
Method__init__
(self, dim_in, dim_out, w0 = 1., c = 6., is_first = False, use_bias = True, activation = None, dropout = False
locationencoder/nn/siren.py:27
Method__init__
(self, dim_in, dim_hidden, dim_out, num_layers, w0 = 1., w0_initial = 30., use_bias = True, final_activation =
locationencoder/nn/siren.py:61
Method__init__
(self, dim_in, dim_hidden, num_layers)
locationencoder/nn/siren.py:104
Method__init__
(self, net, image_width, image_height, latent_dim = None)
locationencoder/nn/siren.py:131
Method__init__
(self)
locationencoder/pe/direct.py:9
Method__init__
(self)
locationencoder/pe/wrap.py:10
Method__init__
legendre_polys: determines the number of legendre polynomials. more polynomials lead more fine-grained resolutions
locationencoder/pe/discretized_spherical_harmonics.py:9
Method__init__
Args: coord_dim: the dimention of space, 2D, 3D, or other frequency_num: the number of different sinusoidal with diff
locationencoder/pe/grid_and_sphere.py:13
Method__init__
Args: coord_dim: the dimention of space, 2D, 3D, or other frequency_num: the number of different sinusoidal with diff
locationencoder/pe/theory.py:19
Method__init__
legendre_polys: determines the number of legendre polynomials. more polynomials lead more fine-grained resolutions
locationencoder/pe/spherical_harmonics.py:7
Method__init__
(self)
locationencoder/pe/cartesian3d.py:10
Method__len__
(self)
data/inat2018_loader.py:155
Methodcal_coord_embed
(self, coords_tuple)
locationencoder/pe/grid_and_sphere.py:61
Functioncompile_summaries
(runsdir,outdir)
experiments/exp_era5.py:373
Methodcompute_features
Compute logits for all images in the dataset using the provided model and store them in logits_file.
data/inat2018_loader.py:180
Methodcompute_logits
Compute logits for all images in the dataset using the provided model and store them in logits_file.
data/inat2018_loader.py:158
Methodconfigure_optimizers
(self)
locationencoder/locationencoder.py:154
Methodconfigure_optimizers
(self)
locationencoder/imageencoder.py:54
Methodconfigure_optimizers
(self)
locationencoder/locationimageencoder.py:72
Functiondefault_loader
(path)
data/inat2018_loader.py:28
Functionfit_comparison
()
experiments/exp_computation.py:13
Functionfit_sh
()
experiments/exp_computation.py:41
Methodforward
(self, images)
locationencoder/imageencoder.py:29
Methodforward
(self, inputs)
locationencoder/locationimageencoder.py:36
Methodforward
(self, x)
locationencoder/nn/fcnet.py:17
Methodforward
(self, x)
locationencoder/nn/fcnet.py:40
Methodforward
(self, x)
locationencoder/nn/siren.py:21
Methodforward
(self, x)
locationencoder/nn/siren.py:51
Methodforward
(self, x, mods = None)
locationencoder/nn/siren.py:85
Methodforward
(self, z)
locationencoder/nn/siren.py:117
Methodforward
(self, img = None, *, latent = None)
locationencoder/nn/siren.py:152
Methodforward
(self, coords)
locationencoder/pe/direct.py:16
Methodforward
(self, lonlat)
locationencoder/pe/discretized_spherical_harmonics.py:37
Methodforward
(self, coords)
locationencoder/pe/grid_and_sphere.py:94
Methodforward
(self, coords)
locationencoder/pe/theory.py:58
Methodforward
(self, lonlat)
locationencoder/pe/spherical_harmonics.py:24
Methodforward
(self, coords)
locationencoder/pe/cartesian3d.py:17
Functionfull_loss
The full "assume negative" loss from Cole and al., 2023, combining SSDL and SLDS. Similar to the Geoprior loss of Mac Aodha et al., 2019, but
utils/losses.py:92
Functionget_air_temp_data
Download and process the Global Air Temperature dataset Parameters: norm_y = logical; should outcome be normalized norm_x = logical;
data/utils.py:78
Functionget_cali_housing_data
Download and process the California Housing Dataset Parameters: norm_x = logical; should features be normalized norm_y = logical; sh
data/utils.py:44
Functionget_election_data
Download and process the Election dataset used in CorrelationGNN (https://arxiv.org/abs/2002.08274) Parameters: norm_x = logical; should
data/utils.py:14
Methodget_weight_matrix
a convenience function to restructure the weight matrix (L x M x E) into a double triangle matrix (L x 2 * L + 1 x E) where with legr
locationencoder/pe/discretized_spherical_harmonics.py:55
Functionobjective
(trial: optuna.trial.Trial)
tune.py:73
Functionpairwise_euclidean_distance
Calculates the pairwise Euclidean distances between each point pair in two matrices. Args: matrix1 (numpy.ndarray): The first matrix
data/checkerboarddataset.py:55
Functionplot_result_matrix
(resultsdir, positional_encoders, neural_networks, show=False, savepath=None)
utils/plot_results.py:226
Methodpredict_step
(self, batch, batch_idx)
locationencoder/locationencoder.py:113
Functionresolution_deg
(N)
data/checkerboarddataset.py:199
Methodsetup
(self, stage: str)
data/checkerboarddataset.py:246
Methodsetup
(self, stage: str)
data/era5dataset.py:89
Methodsetup
(self, stage: str)
data/landoceandataset.py:84
Methodsetup
(self, stage=None)
data/inat2018_loader.py:284
Functionsph2cart
Converts spherical coordinates (longitude, latitude) to 3D Cartesian coordinates (x, y, z). Args: longitude (float or numpy.ndarray)
data/checkerboarddataset.py:38
Methodtest_dataloader
(self)
data/checkerboarddataset.py:266
Methodtest_dataloader
(self)
data/landoceandataset.py:95
Methodtest_dataloader
(self)
data/inat2018_loader.py:307
Methodtest_step
(self, batch, batch_idx)
locationencoder/locationencoder.py:118
Methodtest_step
(self, batch, batch_idx)
locationencoder/imageencoder.py:51
Methodtest_step
(self, batch, batch_idx)
locationencoder/locationimageencoder.py:57
Methodtrain_dataloader
(self)
data/checkerboarddataset.py:260
Methodtrain_dataloader
(self)
data/landoceandataset.py:89
Methodtrain_dataloader
(self)
data/inat2018_loader.py:301
Methodtraining_step
(self, batch, batch_idx)
locationencoder/locationencoder.py:104
Methodtraining_step
(self, batch, batch_idx)
locationencoder/imageencoder.py:32
Methodtraining_step
(self, batch, batch_idx)
locationencoder/locationimageencoder.py:47
Methodval_dataloader
(self)
data/checkerboarddataset.py:263
Methodval_dataloader
(self)
data/era5dataset.py:100
Methodval_dataloader
(self)
data/landoceandataset.py:92
Methodval_dataloader
(self)
data/inat2018_loader.py:304
Methodvalidation_step
(self, batch, batch_idx)
locationencoder/locationencoder.py:108
Methodvalidation_step
(self, batch, batch_idx)
locationencoder/imageencoder.py:37
Methodvalidation_step
(self, batch, batch_idx)
locationencoder/locationimageencoder.py:52
← previous101–175 of 175, ranked by callers