【发布时间】:2019-10-23 02:26:59
【问题描述】:
我有 2000 个不同标签的多类标签文本分类问题。使用 LSTM 和 Glove Embedding 进行分类。
- 目标变量的标签编码器
- 带有嵌入层的 LSTM 层
- 错误度量是 F2 分数
LabelEncoded 目标变量:
le = LabelEncoder()
le.fit(y)
train_y = le.transform(y_train)
test_y = le.transform(y_test)
LSTM 网络如下所示,带有 Glove Embeddings
np.random.seed(seed)
K.clear_session()
model = Sequential()
model.add(Embedding(max_features, embed_dim, input_length = X_train.shape[1],
weights=[embedding_matrix]))#,trainable=False
model.add(LSTM(100, dropout=0.2, recurrent_dropout=0.2))
model.add(Dense(num_classes, activation='softmax'))
model.compile(optimizer='rmsprop', loss='sparse_categorical_crossentropy')
print(model.summary())
我的错误指标是 F1 分数。我为错误指标构建了以下函数
class Metrics(Callback):
def on_train_begin(self, logs={}):
self.val_f1s = []
self.val_recalls = []
self.val_precisions = []
def on_epoch_end(self, epoch, logs={}):
val_predict = (np.asarray(self.model.predict(self.validation_data[0]))).round()
val_targ = self.validation_data[1]
_val_f1 = f1_score(val_targ, val_predict)
_val_recall = recall_score(val_targ, val_predict)
_val_precision = precision_score(val_targ, val_predict)
self.val_f1s.append(_val_f1)
self.val_recalls.append(_val_recall)
self.val_precisions.append(_val_precision)
print("— val_f1: %f — val_precision: %f — val_recall %f" % (_val_f1, _val_precision, _val_recall))
return
metrics = Metrics()
##模型拟合为
model.fit(X_train, train_y, validation_data=(X_test, test_y),epochs=10, batch_size=64, callbacks=[metrics])
在第一个 epoch 后出现以下错误:
ValueError: Classification metrics can't handle a mix of multiclass and continuous-multioutput targets
我的代码哪里出错了?
【问题讨论】:
标签: python machine-learning lstm data-science text-classification