【发布时间】:2019-08-01 04:58:24
【问题描述】:
如何修复此错误?我尝试访问所有论坛以寻找解决此问题的答案。
这里我正在尝试使用 keras 执行多标签分类
from keras.preprocessing.text import Tokenizer
from keras.models import Sequential
from keras.layers import Dense
from keras.preprocessing.sequence import pad_sequences
from keras.layers import Input, Dense, Dropout, Embedding, LSTM, Flatten
from keras.models import Model
from keras.utils import to_categorical
from keras.callbacks import ModelCheckpoint
MAX_LENGTH = 500
tokenizer = Tokenizer()
tokenizer.fit_on_texts(df.overview.values)
post_seq = tokenizer.texts_to_sequences(df.overview.values)
post_seq_padded = pad_sequences(post_seq, maxlen=MAX_LENGTH)
from sklearn.model_selection import train_test_split
X_train, X_test, y_train, y_test = train_test_split(post_seq_padded, train_encoded, test_size=0.3)
vocab_size = len(tokenizer.word_index) + 1
inputs = Input(shape=(MAX_LENGTH, ))
embedding_layer = Embedding(vocab_size, 128, input_length=MAX_LENGTH)(inputs)
x = Dense(64, input_shape=(None,), activation='relu')(embedding_layer)
predictions = Dense(num_classes, activation='softmax')(x)
model = Model(inputs=[inputs], outputs=predictions)
model.compile(optimizer='adam',loss='categorical_crossentropy',metrics=['acc'])
model.summary()
filepath="weights.hdf5"
checkpointer = ModelCheckpoint(filepath, monitor='val_acc', verbose=1, save_best_only=True, mode='max')
history = model.fit(X_train, batch_size=64, y=to_categorical(y_train), verbose=1, validation_split=0.25, shuffle=True, epochs=10, callbacks=[checkpointer])
ValueError Traceback(最近一次调用)
<ipython-input-11-7fdc4bff9648> in <module>
2 checkpointer = ModelCheckpoint(filepath, monitor='val_acc', verbose=1, save_best_only=True, mode='max')
3 history = model.fit(X_train, batch_size=64, y=to_categorical(y_train), verbose=1, validation_split=0.25,
**---->** 4 shuffle=True, epochs=10, callbacks=[checkpointer])
ValueError:检查目标时出错:预期dense_3 的形状为(500, 4),但得到的数组的形状为(4, 2)
我希望输出形状为 (500,3),但我得到的 (4,2) 不匹配以继续进行。
【问题讨论】:
-
你的
df尺寸是多少? -
@Tiendung - 它有 200 行 2 列(它是一个示例数据集)drive.google.com/file/d/1PB7xnodZpT7EFcKq8d6vBmIR9EuczqAm/…
标签: python tensorflow machine-learning keras deep-learning