Fix right padding

hexgnu · Aug 16, 2017 · 43583d3 · 43583d3
1 parent 062a948
commit 43583d3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/ngram.rs b/src/ngram.rs
@@ -39,10 +39,10 @@ impl<'a> NGram<'a> {
     //right-padding
     if !self.pad.is_empty() {
       for i in 1..self.n {
-        let num_blanks = i;
+        let num_blanks = self.n - i;
         let last_entry = tokenized_sequence.len();
         let mut tc = Vec::new();
-        tc.extend_from_slice(&tokenized_sequence[(last_entry - num_blanks) .. last_entry]);
+        tc.extend_from_slice(&tokenized_sequence[(last_entry - i) .. last_entry]);
         for _ in 0..num_blanks {
           tc.push(self.pad);
         }