load_data
函数keras.datasets.reuters.load_data(
path="reuters.npz",
num_words=None,
skip_top=0,
maxlen=None,
test_split=0.2,
seed=113,
start_char=1,
oov_char=2,
index_from=3,
)
加载路透社新闻分类数据集。
这是一个包含 11,228 条路透社新闻的数据集,标记了 46 个主题。
它最初是通过解析和预处理经典的 Reuters-21578 数据集生成的,但预处理代码不再与 Keras 打包在一起。请参阅此 GitHub 讨论 以获取更多信息。
每条新闻都编码为单词索引(整数)列表。为方便起见,单词按其在数据集中出现的整体频率进行索引,例如,整数“3”表示数据中第 3 个最常出现的单词。这允许快速过滤操作,例如:“仅考虑前 10,000 个最常见的单词,但排除前 20 个最常见的单词”。
按照惯例,“0”不代表特定单词,而是用于编码任何未知单词。
参数
~/.keras/dataset
)。num_words
个最常出现的单词。任何频率较低的单词都将作为序列数据中的 oov_char
值出现。如果为 None,则保留所有单词。默认为 None
。oov_char
值出现。0 表示不跳过任何单词。默认为 0
。None
。0.
和 1.
之间的浮点数。用作测试数据的数据集比例。0.2
表示 20% 的数据集用作测试数据。默认为 0.2
。1
。num_words
或 skip_top
限制而被裁剪掉的单词将替换为此字符。返回值
(x_train, y_train), (x_test, y_test)
。x_train
,x_test
:序列列表,它们是索引(整数)列表。如果指定了 num_words 参数,则最大可能的索引值为 num_words - 1
。如果指定了 maxlen
参数,则最大可能的序列长度为 maxlen
。
y_train
,y_test
:整数标签列表(1 或 0)。
注意:“超出词汇表”字符仅用于训练集中存在的但未包含在内的单词,因为它们未达到此处的 num_words
限制。训练集中未见但在测试集中出现的单词已被简单地跳过。
get_word_index
函数keras.datasets.reuters.get_word_index(path="reuters_word_index.json")
检索一个将单词映射到其在路透社数据集中索引的字典。
实际单词索引从 3 开始,其中 3 个索引保留用于:0(填充),1(开始),2(oov)。
例如,“the”的单词索引为 1,但在实际训练数据中,“the”的索引将为 1 + 3 = 4。反之亦然,要使用此映射将训练数据中的单词索引转换回单词,索引需要减去 3。
参数
~/.keras/dataset
)。返回值
单词索引字典。键是单词字符串,值是其索引。