首页 > 解决方案 > 训练具有可变输入大小的 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所以我将不胜感激任何类型的帮助!

标签: pythonkeras

解决方案


推荐阅读