输入大小可变的训练keras模型

时间:2020-10-03 15:49:32

标签: python keras

我正在制造文本分析机。每个向量的大小明显不同,例如:

vectorA = [1, 43, 453, 22, 776, 873, 985]
vectorB = [0, 324, 32432]

我不想确定输入大小,但是我不知道模型如何具有不同的输入大小? 顺便说一句,模型是:

inputs = keras.Input(shape=(None, ))

emb1 = keras.layers.Embedding(
    input_dim=len(X_train), 
    output_dim=None,
    )(inputs)

con1 = keras.layers.Conv1D(
    activation='relu',
    )(emb1)

pool1 = keras.layers.GlobalMaxPooling1D(
    )(con1)

output = keras.layers.Dense(4, 
                           activation="softmax")

model = keras.Model(inputs=inputs, outputs=output)
model.compile(optimizer=keras.optimizers.Adam(),
              loss="binary_crossentropy",
              metrics=["accuracy"])

我是Keras的新手,所以我将不胜感激!

0 个答案:

没有答案