keras中填充整数和字符串的差异

2024-04-25 19:49:51 发布

您现在位置:Python中文网/ 问答频道 /正文

我正在尝试填充一个seq2seq模型的文本。在

from keras_preprocessing.sequence import pad_sequences

x=[["Hello, I'm Bhaskar", "This is Keras"], ["This is an", "experiment"]]
pad_sequences(sequences=x, maxlen=5, dtype='object', padding='pre', value="<PAD>")

我遇到了以下错误:

^{pr2}$

然而,当我尝试对integer做同样的操作时,效果很好。在

x=[[1, 2, 3], [4, 5, 6]]
pad_sequences(sequences=x, maxlen=5, padding='pre', value=0)

Output:
array([[0, 0, 1, 2, 3],
       [0, 0, 4, 5, 6]], dtype=int32)

我希望输出为:

[["<PAD>", "<PAD>", "<PAD>", "Hello, I'm Bhaskar", "This is Keras"], ["<PAD>", "<PAD>","<PAD>", "This is an", "experiment"]]

Tags: anhelloisvaluethisprekerasexperiment
1条回答
网友
1楼 · 发布于 2024-04-25 19:49:51

如错误所示,将dtype改为object(不是字符串而是对象本身),它将为您完成这项工作。在

from keras.preprocessing.sequence import pad_sequences

x=[["Hello, I'm Bhaskar", "This is Keras"], ["This is an", "experiment"]]
pad_sequences(sequences=x, maxlen=5, dtype=object, padding='pre', value="<PAD>")

输出

^{pr2}$

相关问题 更多 >