SYMBOL INDEX (91 symbols across 15 files)

FILE: run.py
  function main (line 92) | def main(argv=None):
  function schedule_sampling (line 115) | def schedule_sampling(eta, itr):
  function train_wrapper (line 158) | def train_wrapper(model):
  function test_wrapper (line 191) | def test_wrapper(model):

FILE: src/data_provider/datasets_factory.py
  function data_provider (line 10) | def data_provider(dataset_name, train_data_paths, valid_data_paths, batc...

FILE: src/data_provider/human.py
  class InputHandle (line 12) | class InputHandle:
    method __init__ (line 13) | def __init__(self, datas, indices, input_param):
    method total (line 26) | def total(self):
    method begin (line 29) | def begin(self, do_shuffle=True):
    method next (line 36) | def next(self):
    method no_batch_left (line 42) | def no_batch_left(self):
    method get_batch (line 48) | def get_batch(self):
    method print_stat (line 68) | def print_stat(self):
  class DataProcess (line 76) | class DataProcess:
    method __init__ (line 77) | def __init__(self, input_param):
    method load_data (line 83) | def load_data(self, paths, mode='train'):
    method get_train_input_handle (line 150) | def get_train_input_handle(self):
    method get_test_input_handle (line 154) | def get_test_input_handle(self):
  function main (line 159) | def main():

FILE: src/data_provider/mnist.py
  class InputHandle (line 4) | class InputHandle:
    method __init__ (line 5) | def __init__(self, input_param):
    method load (line 22) | def load(self):
    method total (line 40) | def total(self):
    method begin (line 43) | def begin(self, do_shuffle = True):
    method next (line 59) | def next(self):
    method no_batch_left (line 74) | def no_batch_left(self):
    method input_batch (line 80) | def input_batch(self):
    method output_batch (line 98) | def output_batch(self):
    method get_batch (line 131) | def get_batch(self):

FILE: src/data_provider/taxibj.py
  function string2timestamp (line 17) | def string2timestamp(strings, T=48):
  class STMatrix (line 30) | class STMatrix(object):
    method __init__ (line 33) | def __init__(self, data, timestamps, T=48, CheckComplete=True):
    method make_index (line 45) | def make_index(self):
    method check_complete (line 50) | def check_complete(self):
    method get_matrix (line 63) | def get_matrix(self, timestamp):
    method save (line 66) | def save(self, fname):
    method check_it (line 69) | def check_it(self, depends):
    method create_dataset (line 75) | def create_dataset(self, len_closeness=20):
  function load_stdata (line 104) | def load_stdata(fname):
  function stat (line 112) | def stat(fname):
  class MinMaxNormalization (line 140) | class MinMaxNormalization(object):
    method __init__ (line 146) | def __init__(self):
    method fit (line 149) | def fit(self, X):
    method transform (line 154) | def transform(self, X):
    method fit_transform (line 159) | def fit_transform(self, X):
    method inverse_transform (line 163) | def inverse_transform(self, X):
  function timestamp2vec (line 169) | def timestamp2vec(timestamps):
  function remove_incomplete_days (line 185) | def remove_incomplete_days(data, timestamps, T=48):
  class InputHandle (line 211) | class InputHandle:
    method __init__ (line 212) | def __init__(self, datas, indices, input_param):
    method total (line 223) | def total(self):
    method begin (line 226) | def begin(self, do_shuffle=True):
    method next (line 233) | def next(self):
    method no_batch_left (line 239) | def no_batch_left(self):
    method get_batch (line 245) | def get_batch(self):
    method print_stat (line 254) | def print_stat(self):
  class DataProcess (line 263) | class DataProcess:
    method __init__ (line 264) | def __init__(self, input_param):
    method load_data (line 274) | def load_data(self, datapath, T=48, nb_flow=2, len_closeness=None, len...
    method get_train_input_handle (line 327) | def get_train_input_handle(self):
    method get_test_input_handle (line 330) | def get_test_input_handle(self):

FILE: src/layers/MIMBlock.py
  class MIMBlock (line 6) | class MIMBlock():
    method __init__ (line 7) | def __init__(self, layer_name, filter_size, num_hidden_in, num_hidden,
    method init_state (line 36) | def init_state(self):
    method MIMS (line 40) | def MIMS(self, x, h_t, c_t):
    method __call__ (line 91) | def __call__(self, x, diff_h, h, c, m):

FILE: src/layers/MIMN.py
  class MIMN (line 4) | class MIMN():
    method __init__ (line 5) | def __init__(self, layer_name, filter_size, num_hidden, seq_shape, tln...
    method init_state (line 26) | def init_state(self):
    method __call__ (line 30) | def __call__(self, x, h_t, c_t):

FILE: src/layers/SpatioTemporalLSTMCellv2.py
  class SpatioTemporalLSTMCell (line 6) | class SpatioTemporalLSTMCell():
    method __init__ (line 7) | def __init__(self, layer_name, filter_size, num_hidden_in, num_hidden,
    method init_state (line 35) | def init_state(self):
    method __call__ (line 39) | def __call__(self, x, h, c, m):

FILE: src/layers/TensorLayerNorm.py
  function tensor_layer_norm (line 6) | def tensor_layer_norm(x, state_name):

FILE: src/models/mim.py
  function w_initializer (line 10) | def w_initializer(dim_in, dim_out):
  function mim (line 15) | def mim(images, params, schedual_sampling_bool, num_layers, num_hidden, ...

FILE: src/models/model_factory.py
  class Model (line 9) | class Model(object):
    method __init__ (line 10) | def __init__(self, configs):
    method train (line 98) | def train(self, inputs, lr, real_input_flag):
    method test (line 105) | def test(self, inputs, real_input_flag):
    method save (line 111) | def save(self, itr):
    method load (line 116) | def load(self, checkpoint_path):
    method construct_model (line 120) | def construct_model(self, name, images, model_params, real_input_flag,...

FILE: src/trainer.py
  function train (line 10) | def train(model, ims, real_input_flag, configs, itr, ims_reverse=None):
  function test (line 38) | def test(model, test_input_handle, configs, save_name):

FILE: src/utils/metrics.py
  function batch_mae_frame_float (line 7) | def batch_mae_frame_float(gen_frames, gt_frames):
  function batch_psnr (line 19) | def batch_psnr(gen_frames, gt_frames):

FILE: src/utils/optimizer.py
  function adam_updates (line 4) | def adam_updates(params, cost_or_grads, lr=0.001, mom1=0.9, mom2=0.999):

FILE: src/utils/preprocess.py
  function reshape_patch (line 6) | def reshape_patch(img_tensor, patch_size):
  function reshape_patch_back (line 25) | def reshape_patch_back(patch_tensor, patch_size):