tf.keras.datasets.imdb.load_data
| Просмотреть исходный код на GitHub |
Загружает набор данных IMDB.
tf.keras.datasets.imdb.load_data(
path='imdb.npz', num_words=None, skip_top=0, maxlen=None, seed=113,
start_char=1, oov_char=2, index_from=3, **kwargs
)
Это набор данных из 25 000 обзоров фильмов IMDB, помеченных по эмоциональной окраске (положительная/отрицательная). Обзоры были предварительно обработаны, и каждый обзор закодирован как список индексов слов (целые числа). Для удобства слова индексируются по общей частоте в наборе данных, так что, например, целое число «3» кодирует третье по частоте слово в данных. Это позволяет быстро выполнять операции фильтрации, такие как: «рассмотреть только 10 000 самых распространенных слов, но исключить 20 самых распространенных слов».
По соглашению, «0» не обозначает конкретное слово, а используется для кодирования любого неизвестного слова.
| Аргументы | |
|---|---|
path | место для кэширования данных (относительно ~/.keras/dataset). |
num_words | целое число или None. Слова ранжируются по частоте (в обучающей выборке), и сохраняются только num_words самых часто встречающихся слов. Любое менее частое слово будет отображаться как значение oov_char в данных последовательности. Если None, сохраняются все слова. По умолчанию None, поэтому сохраняются все слова. |
skip_top | пропустить первые N наиболее часто встречающихся слов (которые могут быть неинформативными). Эти слова будут отображаться как значение oov_char в наборе данных. По умолчанию 0, поэтому слова не пропускаются. |
maxlen | целое число или None. Максимальная длина последовательности. Любая более длинная последовательность будет усечена. По умолчанию None, что означает отсутствие усечения. |
seed | целое число. Зерно для воспроизводимой перестановки данных. |
start_char | целое число. Начало последовательности будет помечено этим символом. По умолчанию 1, так как 0 обычно является символом заполнения. |
oov_char | целое число. Символ слова вне словаря. Слова, которые были исключены из-за ограничений num_words или skip_top будут заменены этим символом. |
index_from | целое число. Индексировать фактические слова с этим индексом и выше. |
**kwargs | Используется для обратной совместимости. |
| Возвращаемое значение | |
|---|---|
Кортеж массивов NumPy: (x_train, y_train), (x_test, y_test). x_train, x_test: списки последовательностей, которые являются списками индексов (целых чисел). Если был указан аргумент num_words, максимальное возможное значение индекса — y_train, y_test: списки целых меток (1 или 0). |
| Исключения | |
|---|---|
ValueError | в случае, если значение maxlen настолько низкое, что ни одна входная последовательность не может быть сохранена. |
Обратите внимание, что символ «слово вне словаря» используется только для слов, которые присутствовали в обучающей выборке, но не включены, потому что они не прошли порог num_words здесь. Слова, которые не встречались в обучающей выборке, но присутствуют в тестовой выборке, были просто пропущены.
© 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.3/api_docs/python/tf/keras/datasets/imdb/load_data
