def get_equivalent_ids(self, split_1="train", split_2="test"):
    """Return a list of equivalent ids between two splits."""
    equivalent_ids = []
    example_map = collections.defaultdict(int)
    for i, example in enumerate(self._data_pairs[split_1]):
      template_identifier = self._template_identifiers[split_1][i]
      command, target_command, row, column = self.get_example_info(example)
      key = (template_identifier, command, target_command, row, column)
      example_map[key] += 1
    for i, example in enumerate(self._data_pairs[split_2]):
      template_identifier = self._template_identifiers[split_2][i]
      command, target_command, row, column = self.get_example_info(example)
      key = (template_identifier, command, target_command, row, column)
      if example_map[key] > 0:
        equivalent_ids.append(i)
    return equivalent_ids
Return a list of equivalent ids between two splits.
language/gscan/data/dataset.py