models/research/sentiment_analysis/data/util.py

"""Utility module for sentiment analysis."""

from __future__ import absolute_import
from __future__ import division
from __future__ import print_function

import numpy as np

START_CHAR = 1
END_CHAR = 2
OOV_CHAR = 3


def pad_sentence(sentence, sentence_length):
  """Pad the given sentense at the end.

  If the input is longer than sentence_length,
  the remaining portion is dropped.
  END_CHAR is used for the padding.

  Args:
    sentence: A numpy array of integers.
    sentence_length: The length of the input after the padding.
  Returns:
    A numpy array of integers of the given length.
  """
  sentence = sentence[:sentence_length]
  if len(sentence) < sentence_length:
    sentence = np.pad(sentence, (0, sentence_length - len(sentence)),
                      "constant", constant_values=(START_CHAR, END_CHAR))

  return sentence