【发布时间】:2020-10-05 03:06:59
【问题描述】:
我正在为使用 KerasClassifier 模型的多类问题构建链分类器。我有 17 个标签作为分类目标,X_train 的形状是 (111300,107),y_train 是 (111300,17) 我的代码在这里:
def create_model():
input_size=length_long_sentence
embedding_size=128
lstm_size=64
output_size=len(unique_tag_set)
#----------------------------Model -------------------------------
current_input=Input(shape=(input_size,))
emb_current = Embedding(vocab_size, embedding_size, input_length=input_size)(current_input)
out_current=Bidirectional(LSTM(units=lstm_size))(emb_current )
#out_current = Reshape((1,2*lstm_size))(out_current)
output = Dense(units=len(unique_tag_set), activation='softmax')(out_current)
model = Model(inputs=current_input, outputs=output)
model.compile(optimizer='Adam', loss='categorical_crossentropy', metrics=['accuracy'])
print(model.summary())
return model
model = KerasClassifier(build_fn=create_model, epochs=1,batch_size=256)
print(type(model))
chain=ClassifierChain(model, order='random', random_state=42)
history=chain.fit(X_train, y_train)
模型摘要在这里:
当尝试在 ClassifierChain 上使用 fit 方法时,我收到此错误:
任何人都可以指导我这个错误以及什么是 (None,2)?
【问题讨论】:
-
我认为softmax不适合多标签设置,chainclassifier就是这种情况。
标签: python keras scikit-learn multilabel-classification