SYMBOL INDEX (220 symbols across 36 files)

FILE: analysis/eval_intrinsic_dimension.py
  function load_config (line 13) | def load_config(filepath):
  function remove_duplicates (line 21) | def remove_duplicates(X):
  function eval_id_latent (line 25) | def eval_id_latent(vars_filepath, if_refine, if_all_methods):
  function eval_id_image (line 45) | def eval_id_image(data_filepath, test_vid_ids, vars_filepath, if_all_met...

FILE: analysis/eval_phys_data.py
  function eval_phys_data_single_pendulum (line 8) | def eval_phys_data_single_pendulum(data_filepath, num_vids, num_frms, sa...
  function eval_phys_data_double_pendulum (line 27) | def eval_phys_data_double_pendulum(data_filepath, num_vids, num_frms, sa...
  function eval_phys_data_elastic_pendulum (line 58) | def eval_phys_data_elastic_pendulum(data_filepath, num_vids, num_frms, s...

FILE: analysis/eval_phys_double_pendulum/__init__.py
  function eval_physics (line 11) | def eval_physics(frames):

FILE: analysis/eval_phys_double_pendulum/angle_estimator.py
  function seg_from_img (line 22) | def seg_from_img(img):
  function fit_pendulum (line 50) | def fit_pendulum(seg):
  function estimate_angle (line 85) | def estimate_angle(rect, ref_pt):
  function obtain_angle (line 119) | def obtain_angle(img):

FILE: analysis/eval_phys_double_pendulum/physics_estimator.py
  function calc_diff (line 20) | def calc_diff(th1, th2):
  function calc_avrg (line 29) | def calc_avrg(th1, th2):
  function calc_velocity (line 44) | def calc_velocity(th, method='spline'):
  function calc_energy (line 83) | def calc_energy(th1, th2, vel_th1, vel_th2):

FILE: analysis/eval_phys_elastic_pendulum/__init__.py
  function eval_physics (line 11) | def eval_physics(frames):

FILE: analysis/eval_phys_elastic_pendulum/angle_estimator.py
  function seg_from_img (line 22) | def seg_from_img(img):
  function fit_pendulum (line 50) | def fit_pendulum(seg):
  function estimate_angle (line 85) | def estimate_angle(rect, ref_pt):
  function obtain_angle_stretch (line 120) | def obtain_angle_stretch(img):

FILE: analysis/eval_phys_elastic_pendulum/physics_estimator.py
  function calc_diff (line 22) | def calc_diff(th1, th2):
  function calc_avrg (line 31) | def calc_avrg(th1, th2):
  function normalize_angle (line 43) | def normalize_angle(theta):
  function calc_velocity (line 52) | def calc_velocity(x, method='spline', periodic=True):
  function calc_energy (line 92) | def calc_energy(th1, th2, z, vel_th1, vel_th2, vel_z):

FILE: analysis/eval_phys_long_term_pred.py
  function calc_diff (line 21) | def calc_diff(th1, th2):
  function calc_pixel_MSE (line 29) | def calc_pixel_MSE(img1, img2):
  function load_config (line 32) | def load_config(filepath):
  function eval_pred_physics (line 41) | def eval_pred_physics(data_filepath, pred_save_path, vid_ids, phys_vars_...
  function load_data_physics (line 69) | def load_data_physics(data_filepath, vid_ids, phys_vars_list):
  function eval_physics_error (line 76) | def eval_physics_error(phys_pred, phys_data, phys_vars_list):
  function eval_pixel_error (line 95) | def eval_pixel_error(data_filepath, pred_save_path, vid_ids):
  function main (line 117) | def main():

FILE: analysis/eval_phys_single_pendulum/__init__.py
  function eval_physics (line 11) | def eval_physics(frames):

FILE: analysis/eval_phys_single_pendulum/angle_estimator.py
  function seg_from_img (line 21) | def seg_from_img(img):
  function fit_pendulum (line 40) | def fit_pendulum(seg):
  function estimate_angle (line 74) | def estimate_angle(rect):
  function obtain_angle (line 108) | def obtain_angle(img):

FILE: analysis/eval_phys_single_pendulum/physics_estimator.py
  function calc_diff (line 20) | def calc_diff(th1, th2):
  function calc_avrg (line 29) | def calc_avrg(th1, th2):
  function normalize_angle (line 41) | def normalize_angle(theta):
  function calc_velocity (line 50) | def calc_velocity(th, method='spline'):
  function calc_energy (line 90) | def calc_energy(th, vel_th):

FILE: analysis/eval_regression.py
  function load_config (line 12) | def load_config(filepath):
  function parse_data_id (line 22) | def parse_data_id(data_id):
  function physical_variables_from_data_ids (line 29) | def physical_variables_from_data_ids(phys_all, ids):
  function main (line 51) | def main():

FILE: analysis/intrinsic_dimension_estimation/__init__.py
  class ID_Estimator (line 4) | class ID_Estimator:
    method __init__ (line 5) | def __init__(self, method='Levina_Bickel'):
    method set_method (line 9) | def set_method(self, method='Levina_Bickel'):
    method fit (line 15) | def fit(self, X, k_list=20, n_jobs=4):
    method fit_all_methods (line 41) | def fit_all_methods(self, X, k_list=[20], n_jobs=4):

FILE: analysis/intrinsic_dimension_estimation/methods.py
  function kNN (line 6) | def kNN(X, n_neighbors, n_jobs):
  function Levina_Bickel (line 12) | def Levina_Bickel(X, dists, k):
  function start_matlab_engine (line 19) | def start_matlab_engine():
  function MiND_ML (line 26) | def MiND_ML(X, dists, k):
  function MiND_KL (line 35) | def MiND_KL(X, dists, k, maxDim=30):
  function DANCo (line 45) | def DANCo(X, dists, inds, k, maxDim=30):
  function Hein_CD (line 56) | def Hein_CD(X):

FILE: analysis/latent_regression/regressors.py
  function mlp_regress (line 8) | def mlp_regress(X_train, X_test, y_train, y_test, random_state=0):
  function lin_regress (line 21) | def lin_regress(X_train, X_test, y_train, y_test):
  function pca (line 29) | def pca(X_train, X_test, num_components, random_state=0):

FILE: datainfo/collect/circular_motion/make_data.py
  function mkdir (line 7) | def mkdir(folder):
  function coord2img (line 21) | def coord2img(x, y):

FILE: datainfo/collect/double_pendulum/convert_video.py
  function mkdir (line 13) | def mkdir(folder):
  function get_arguments (line 19) | def get_arguments():
  function write_frames (line 30) | def write_frames(video_path, frame_path, fps, seed):

FILE: datainfo/collect/double_pendulum/equalize_background.py
  function mkdir (line 15) | def mkdir(folder):
  function get_arguments (line 21) | def get_arguments():

FILE: datainfo/collect/double_pendulum/split_data.py
  function mkdir (line 12) | def mkdir(folder):
  function get_arguments (line 18) | def get_arguments():

FILE: datainfo/collect/elastic_pendulum/make_data.py
  function mkdir (line 9) | def mkdir(folder):
  function engine (line 15) | def engine(rng, num_frm, fps=60):
  function draw_rect (line 58) | def draw_rect(im, col, top_x, top_y, w, h, theta):
  function render (line 73) | def render(theta_1, theta_2, z):
  function make_data (line 95) | def make_data(data_filepath, num_seq, num_frm, seed=0):

FILE: datainfo/collect/fire/split_data.py
  function mkdir (line 12) | def mkdir(folder):
  function get_arguments (line 18) | def get_arguments():

FILE: datainfo/collect/lava_lamp/split_data.py
  function mkdir (line 12) | def mkdir(folder):
  function get_arguments (line 18) | def get_arguments():

FILE: datainfo/collect/reaction_diffusion/make_data.py
  function mkdir (line 10) | def mkdir(folder):
  function make_data (line 16) | def make_data(data_filepath, num_frm):

FILE: datainfo/collect/single_pendulum/make_data.py
  function mkdir (line 8) | def mkdir(folder):
  function engine (line 14) | def engine(rng, num_frm, fps=60):
  function render (line 32) | def render(theta):
  function make_data (line 42) | def make_data(data_filepath, num_seq, num_frm, seed=0):

FILE: datainfo/collect/utils/sort_vids.py
  function mkdir (line 10) | def mkdir(folder):

FILE: dataset.py
  class NeuralPhysDataset (line 14) | class NeuralPhysDataset(Dataset):
    method __init__ (line 15) | def __init__(self, data_filepath, flag, seed, object_name="double_pend...
    method get_all_filelist (line 22) | def get_all_filelist(self):
    method __len__ (line 42) | def __len__(self):
    method __getitem__ (line 46) | def __getitem__(self, idx):
    method get_data (line 59) | def get_data(self, filepath):
  class NeuralPhysLatentDynamicsDataset (line 69) | class NeuralPhysLatentDynamicsDataset(Dataset):
    method __init__ (line 70) | def __init__(self, data_filepath, flag, seed, object_name="double_pend...
    method get_all_filelist (line 77) | def get_all_filelist(self):
    method __len__ (line 97) | def __len__(self):
    method __getitem__ (line 101) | def __getitem__(self, idx):
    method get_data (line 118) | def get_data(self, filepath):

FILE: eval.py
  function mkdir (line 21) | def mkdir(folder):
  function load_config (line 26) | def load_config(filepath):
  function seed (line 34) | def seed(cfg):
  function main (line 40) | def main():
  function main_latentpred (line 97) | def main_latentpred():
  function rename_ckpt_for_multi_models (line 144) | def rename_ckpt_for_multi_models(ckpt):
  function gather_latent_from_trained_high_dim_model (line 155) | def gather_latent_from_trained_high_dim_model():
  function gather_latent_from_trained_refine_model (line 253) | def gather_latent_from_trained_refine_model():

FILE: main.py
  function load_config (line 15) | def load_config(filepath):
  function seed (line 23) | def seed(cfg):
  function main (line 29) | def main():
  function main_latentpred (line 77) | def main_latentpred():

FILE: model_utils.py
  function conv2d_bn_relu (line 7) | def conv2d_bn_relu(inch,outch,kernel_size,stride=1,padding=1):
  function conv2d_bn_sigmoid (line 15) | def conv2d_bn_sigmoid(inch,outch,kernel_size,stride=1,padding=1):
  function deconv_sigmoid (line 23) | def deconv_sigmoid(inch,outch,kernel_size,stride=1,padding=1):
  function deconv_relu (line 30) | def deconv_relu(inch,outch,kernel_size,stride=1,padding=1):
  class EncoderDecoder (line 40) | class EncoderDecoder(torch.nn.Module):
    method __init__ (line 41) | def __init__(self, in_channels):
    method encoder (line 94) | def encoder(self, x):
    method decoder (line 102) | def decoder(self, x):
    method forward (line 123) | def forward(self,x):
  class EncoderDecoder64x1x1 (line 128) | class EncoderDecoder64x1x1(torch.nn.Module):
    method __init__ (line 129) | def __init__(self, in_channels):
    method encoder (line 220) | def encoder(self, x):
    method decoder (line 231) | def decoder(self, x):
    method forward (line 265) | def forward(self,x, reconstructed_latent, refine_latent):
  class SirenLayer (line 275) | class SirenLayer(nn.Module):
    method __init__ (line 276) | def __init__(self, in_f, out_f, w0=30, is_first=False, is_last=False):
    method init_weights (line 285) | def init_weights(self):
    method forward (line 290) | def forward(self, x):
  class RefineDoublePendulumModel (line 294) | class RefineDoublePendulumModel(torch.nn.Module):
    method __init__ (line 295) | def __init__(self, in_channels):
    method forward (line 307) | def forward(self, x):
  class RefineElasticPendulumModel (line 318) | class RefineElasticPendulumModel(torch.nn.Module):
    method __init__ (line 319) | def __init__(self, in_channels):
    method forward (line 331) | def forward(self, x):
  class RefineReactionDiffusionModel (line 342) | class RefineReactionDiffusionModel(torch.nn.Module):
    method __init__ (line 343) | def __init__(self, in_channels):
    method forward (line 355) | def forward(self, x):
  class RefineSwingStickNonMagneticModel (line 366) | class RefineSwingStickNonMagneticModel(torch.nn.Module):
    method __init__ (line 367) | def __init__(self, in_channels):
    method forward (line 379) | def forward(self, x):
  class RefineSinglePendulumModel (line 390) | class RefineSinglePendulumModel(torch.nn.Module):
    method __init__ (line 391) | def __init__(self, in_channels):
    method forward (line 403) | def forward(self, x):
  class RefineCircularMotionModel (line 415) | class RefineCircularMotionModel(torch.nn.Module):
    method __init__ (line 416) | def __init__(self, in_channels):
    method forward (line 428) | def forward(self, x):
  class RefineAirDancerModel (line 439) | class RefineAirDancerModel(torch.nn.Module):
    method __init__ (line 440) | def __init__(self, in_channels):
    method forward (line 452) | def forward(self, x):
  class RefineLavaLampModel (line 463) | class RefineLavaLampModel(torch.nn.Module):
    method __init__ (line 464) | def __init__(self, in_channels):
    method forward (line 476) | def forward(self, x):
  class RefineFireModel (line 487) | class RefineFireModel(torch.nn.Module):
    method __init__ (line 488) | def __init__(self, in_channels):
    method forward (line 500) | def forward(self, x):
  class RefineModelReLU (line 512) | class RefineModelReLU(torch.nn.Module):
    method __init__ (line 513) | def __init__(self, in_channels):
    method forward (line 528) | def forward(self, x):
  class LatentPredModel (line 538) | class LatentPredModel(torch.nn.Module):
    method __init__ (line 539) | def __init__(self, in_channels):
    method forward (line 554) | def forward(self, x):

FILE: models.py
  function mkdir (line 29) | def mkdir(folder):
  class VisDynamicsModel (line 34) | class VisDynamicsModel(pl.LightningModule):
    method __init__ (line 36) | def __init__(self,
    method __build_model (line 63) | def __build_model(self):
    method train_forward (line 92) | def train_forward(self, x):
    method training_step (line 99) | def training_step(self, batch, batch_idx):
    method validation_step (line 107) | def validation_step(self, batch, batch_idx):
    method test_step (line 115) | def test_step(self, batch, batch_idx):
    method test_save (line 173) | def test_save(self):
    method configure_optimizers (line 183) | def configure_optimizers(self):
    method paths_to_tuple (line 188) | def paths_to_tuple(self, paths):
    method setup (line 195) | def setup(self, stage=None):
    method train_dataloader (line 234) | def train_dataloader(self):
    method val_dataloader (line 241) | def val_dataloader(self):
    method test_dataloader (line 248) | def test_dataloader(self):

FILE: models_latentpred.py
  function mkdir (line 31) | def mkdir(folder):
  function rename_ckpt_for_multi_models (line 36) | def rename_ckpt_for_multi_models(ckpt):
  class VisLatentDynamicsModel (line 44) | class VisLatentDynamicsModel(pl.LightningModule):
    method __init__ (line 46) | def __init__(self,
    method __build_model (line 73) | def __build_model(self):
    method load_model (line 110) | def load_model(self, checkpoint_filepath):
    method load_high_dim_refine_model (line 121) | def load_high_dim_refine_model(self, high_dim_checkpoint_filepath, ref...
    method extract_decoder_from_refine_model (line 141) | def extract_decoder_from_refine_model(self):
    method data_to_state (line 149) | def data_to_state(self, data):
    method training_step (line 156) | def training_step(self, batch, batch_idx):
    method validation_step (line 166) | def validation_step(self, batch, batch_idx):
    method test_step (line 176) | def test_step(self, batch, batch_idx):
    method configure_optimizers (line 204) | def configure_optimizers(self):
    method setup (line 209) | def setup(self, stage=None):
    method train_dataloader (line 226) | def train_dataloader(self):
    method val_dataloader (line 233) | def val_dataloader(self):
    method test_dataloader (line 240) | def test_dataloader(self):

FILE: pred.py
  function mkdir (line 39) | def mkdir(folder):
  function load_config (line 44) | def load_config(filepath):
  function seed (line 52) | def seed(cfg):
  function model_rollout (line 60) | def model_rollout():
  function model_rollout_hybrid (line 213) | def model_rollout_hybrid(step):
  function model_rollout_perturb (line 319) | def model_rollout_perturb(perturb_type, perturb_level):
  function rename_ckpt_for_multi_models (line 480) | def rename_ckpt_for_multi_models(ckpt):
  function get_data (line 490) | def get_data(filepath):
  function get_data_perturb (line 498) | def get_data_perturb(filepath, perturb_type, perturb_level):
  function tensor_to_img (line 527) | def tensor_to_img(out_tensor):

FILE: stability.py
  function mkdir (line 19) | def mkdir(folder):
  function load_config (line 24) | def load_config(filepath):
  function seed (line 32) | def seed(cfg):
  function get_data (line 38) | def get_data(filepath):
  function main (line 47) | def main():

FILE: utils/common.py
  function mkdir (line 10) | def mkdir(folder):
  function create_random_data_splits (line 16) | def create_random_data_splits(seed, data_filepath, object_name, ratio=0.8):
  function separate_eval_image (line 39) | def separate_eval_image(filepath):
  function collect_long_term_pred_sample (line 49) | def collect_long_term_pred_sample(dataset, seed, test_vid_id, hybrid_ste...
  function light_color (line 99) | def light_color(col):
  function update_figure (line 106) | def update_figure(fig):
  function test (line 119) | def test():

FILE: utils/dimension.py
  function load_config (line 8) | def load_config(filepath):
  function calculate_intrinsic_dimension_statistics (line 17) | def calculate_intrinsic_dimension_statistics(dataset, model_type='model'...
  function calculate_intrinsic_dimension_statistics_all_methods (line 38) | def calculate_intrinsic_dimension_statistics_all_methods(dataset, model_...