def _load_fever_embeddings(self):
    """Load fever claim embeddings and ids."""
    with util.safe_open(self._claim_embedding_path, 'rb') as f:
      claim_embeddings = np.load(f)
    with util.safe_open(self._claim_id_path, 'rb') as f:
      claim_ids = np.load(f)
    self._claim_embeddings = tf.convert_to_tensor(claim_embeddings)
    if self._l2_norm:
      self._claim_embeddings = tf.math.l2_normalize(
          self._claim_embeddings, axis=-1)
    self._claim_ids = claim_ids
    self._claim_id_to_idx = {}
    for idx, claim_id in enumerate(self._claim_ids):
      self._claim_id_to_idx[claim_id] = idx
Load fever claim embeddings and ids.
language/serene/wiki_index.py