SYMBOL INDEX (32 symbols across 4 files)

FILE: layers.py
  function to_npy (line 13) | def to_npy(x):
  class Attn (line 17) | class Attn(nn.Module):
    method __init__ (line 18) | def __init__(self, emb_loc, loc_max, dropout=0.1):
    method forward (line 24) | def forward(self, self_attn, self_delta, traj_len):
  class SelfAttn (line 39) | class SelfAttn(nn.Module):
    method __init__ (line 40) | def __init__(self, emb_size, output_size, dropout=0.1):
    method forward (line 46) | def forward(self, joint, delta, traj_len):
  class Embed (line 62) | class Embed(nn.Module):
    method __init__ (line 63) | def __init__(self, ex, emb_size, loc_max, embed_layers):
    method forward (line 70) | def forward(self, traj_loc, mat2, vec, traj_len):
  class MultiEmbed (line 94) | class MultiEmbed(nn.Module):
    method __init__ (line 95) | def __init__(self, ex, emb_size, embed_layers):
    method forward (line 102) | def forward(self, traj, mat, traj_len):

FILE: load.py
  function haversine (line 10) | def haversine(lon1, lat1, lon2, lat2):
  function euclidean (line 25) | def euclidean(point, each):
  function rst_mat1 (line 30) | def rst_mat1(traj, poi):
  function rs_mat2s (line 41) | def rs_mat2s(poi, l_max):
  function rt_mat2t (line 53) | def rt_mat2t(traj_time):  # traj_time (*M+1) triangle matrix
  function process_traj (line 64) | def process_traj(dname):  # start from 1

FILE: models.py
  class Model (line 4) | class Model(nn.Module):
    method __init__ (line 5) | def __init__(self, t_dim, l_dim, u_dim, embed_dim, ex, dropout=0.1):
    method forward (line 21) | def forward(self, traj, mat1, mat2, vec, traj_len):

FILE: train.py
  function calculate_acc (line 10) | def calculate_acc(prob, label):
  function sampling_prob (line 24) | def sampling_prob(prob, label, num_neg):
  class DataSet (line 49) | class DataSet(data.Dataset):
    method __init__ (line 50) | def __init__(self, traj, m1, v, label, length):
    method __getitem__ (line 54) | def __getitem__(self, index):
    method __len__ (line 62) | def __len__(self):  # no use
  class Trainer (line 66) | class Trainer:
    method __init__ (line 67) | def __init__(self, model, record):
    method train (line 86) | def train(self):
    method inference (line 161) | def inference(self):