def get_oracle_neighbors(self, q_id):
    """Look-up the neighbors in the oracle table."""
    result = []
    if not self.oracle_neighbors:
      return result
    neighbors = self.oracle_neighbors[q_id]

    for n in range(len(neighbors)):
      next_n = neighbors[n]
      result.append(next_n[0])
    return result
Look-up the neighbors in the oracle table.
language/multivec/preprocessing/create_training_data.py