Skip to content

Commit

Permalink
Merge branch 'master' into patch-2
Browse files Browse the repository at this point in the history
  • Loading branch information
IlyaGusev authored May 22, 2024
2 parents ac9687d + d05dcbe commit 9954c48
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions rnnmorph/model.py
Original file line number Diff line number Diff line change
Expand Up @@ -376,10 +376,10 @@ def predict_probabilities(self, sentences: List[List[str]], batch_size: int,
return [[] for _ in sentences]
n_samples = len(sentences)

words = np.zeros((n_samples, max_sentence_len), dtype=np.int)
words = np.zeros((n_samples, max_sentence_len), dtype=int)
grammemes = np.zeros((n_samples, max_sentence_len, self.grammeme_vectorizer_input.grammemes_count()),
dtype=float)
chars = np.zeros((n_samples, max_sentence_len, build_config.char_max_word_length), dtype=np.int)
chars = np.zeros((n_samples, max_sentence_len, build_config.char_max_word_length), dtype=int)

for i, sentence in enumerate(sentences):
if not sentence:
Expand Down

0 comments on commit 9954c48

Please sign in to comment.