tf.keras.preprocessing.text.one_hot
View source on GitHub |
One-hot encodes a text into a list of word indexes of size n
.
tf.keras.preprocessing.text.one_hot( input_text, n, filters='!"#$%&()*+,-./:;<=>?@[\\]^_`{|}~\t\n', lower=True, split=' ' )
This function receives as input a string of text and returns a list of encoded integers each corresponding to a word (or token) in the given input string.
Arguments | |
---|---|
input_text | Input text (string). |
n | int. Size of vocabulary. |
filters | list (or concatenation) of characters to filter out, such as punctuation. Default: '!"#$%&()*+,-./:;<=>?@[\]^_`{|}~\t\n , includes basic punctuation, tabs, and newlines. |
lower | boolean. Whether to set the text to lowercase. |
split | str. Separator for word splitting. |
Returns | |
---|---|
List of integers in [1, n] . Each integer encodes a word (unicity non-guaranteed). |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/keras/preprocessing/text/one_hot