What does Keras Tokenizer method exactly do?
nuric already satistified the question, but I would add something.
Please focus on both word frequency-based encoding and OOV in this example:
from tensorflow.keras.preprocessing.text import Tokenizer
corpus =['The', 'cat', 'is', 'on', 'the', 'table', 'a', 'very', 'long', 'table']
tok_obj = Tokenizer(num_words=10, oov_token='<OOV>')
tok_obj.fit_on_texts(corpus)
[TL;DR] The tokenizer will include the first 10
words appearing in the corpus. Here 10
words, but only 8 are unique. The most frequent 10
words will be encoded, if they are more than this number they will go OOV (Out Of Vocabulary).
Built dictionary:
Please note the frequency
{'<OOV>': 1, 'the': 2, 'table': 3, 'cat': 4, 'is': 5, 'on': 6, 'a': 7, 'very': 8, 'long': 9}
Sentence(s) processing:
processed_seq = tok_obj.texts_to_sequences(['The dog is on the bed'])
Which gives:
>>> processed_seq
[[2, 1, 5, 6, 2, 1]]
How to retrieve the sentence?
Build the dictionary inv_map
and use It! list comprehension can be used below to compress the code.
inv_map = {v: k for k, v in tok_obj.word_index.items()}
for seq in processed_seq:
for tok in seq:
print(inv_map[tok])
which gives:
>>> the
<OOV>
is
on
the
<OOV>
because dog and bed are not in the dictionary.
List comprehension can be used to compress the code. Here obtaining a list as output.
[inv_map[tok] for seq in processed_seq for tok in seq]
which gives:
>>> ['the', '<OOV>', 'is', 'on', 'the', '<OOV>']
Adding more to above answers with examples will help in better understanding:
Example 1:
t = Tokenizer()
fit_text = "The earth is an awesome place live"
t.fit_on_texts(fit_text)
test_text = "The earth is an great place live"
sequences = t.texts_to_sequences(test_text)
print("sequences : ",sequences,'\n')
print("word_index : ",t.word_index)
#[] specifies : 1. space b/w the words in the test_text 2. letters that have not occured in fit_text
Output :
sequences : [[3], [4], [1], [], [1], [2], [8], [3], [4], [], [5], [6], [], [2], [9], [], [], [8], [1], [2], [3], [], [13], [7], [2], [14], [1], [], [7], [5], [15], [1]]
word_index : {'e': 1, 'a': 2, 't': 3, 'h': 4, 'i': 5, 's': 6, 'l': 7, 'r': 8, 'n': 9, 'w': 10, 'o': 11, 'm': 12, 'p': 13, 'c': 14, 'v': 15}
Example 2:
t = Tokenizer()
fit_text = ["The earth is an awesome place live"]
t.fit_on_texts(fit_text)
#fit_on_texts fits on sentences when list of sentences is passed to fit_on_texts() function.
#ie - fit_on_texts( [ sent1, sent2, sent3,....sentN ] )
#Similarly, list of sentences/single sentence in a list must be passed into texts_to_sequences.
test_text1 = "The earth is an great place live"
test_text2 = "The is my program"
sequences = t.texts_to_sequences([test_text1, test_text2])
print('sequences : ',sequences,'\n')
print('word_index : ',t.word_index)
#texts_to_sequences() returns list of list. ie - [ [] ]
Output:
sequences : [[1, 2, 3, 4, 6, 7], [1, 3]]
word_index : {'the': 1, 'earth': 2, 'is': 3, 'an': 4, 'awesome': 5, 'place': 6, 'live': 7}
Lets see what this line of code does.
tokenizer.fit_on_texts(text)
For example, consider the sentence " The earth is an awesome place live"
tokenizer.fit_on_texts("The earth is an awesome place live")
fits [[1,2,3,4,5,6,7]]
where 3 -> "is" , 6 -> "place", so on.
sequences = tokenizer.texts_to_sequences("The earth is an great place live")
returns [[1,2,3,4,6,7]]
.
You see what happened here. The word "great" is not fit initially, so it does not recognize the word "great". Meaning, fit_on_text can be used independently on train data and then the fitted vocabulary index can be used to represent a completely new set of word sequence. These are two different processes. Hence the two lines of code.
From the source code:
fit_on_texts
Updates internal vocabulary based on a list of texts. This method creates the vocabulary index based on word frequency. So if you give it something like, "The cat sat on the mat." It will create a dictionary s.t.word_index["the"] = 1; word_index["cat"] = 2
it is word -> index dictionary so every word gets a unique integer value. 0 is reserved for padding. So lower integer means more frequent word (often the first few are stop words because they appear a lot).texts_to_sequences
Transforms each text in texts to a sequence of integers. So it basically takes each word in the text and replaces it with its corresponding integer value from theword_index
dictionary. Nothing more, nothing less, certainly no magic involved.
Why don't combine them? Because you almost always fit once and convert to sequences many times. You will fit on your training corpus once and use that exact same word_index
dictionary at train / eval / testing / prediction time to convert actual text into sequences to feed them to the network. So it makes sense to keep those methods separate.