我正在尝试使用RNN和CRF为NER生成模型。
但是我遇到错误,并且由于错误原因找不到任何信息。如果有人可以帮助我,那真是太好了。
”使用输入暗淡2的索引超出范围;输入只有2个暗淡 [[{{node loss_15 / crf_17_loss / strided_slice_4}}]]“
model = Sequential()
model.add(Embedding(input_dim=wordEmbeddings.shape[0], output_dim=wordEmbeddings.shape[1],
input_length=n_in, weights=[wordEmbeddings], trainable=False))
crf = CRF(n_out, sparse_target=False)
model.add(crf)
model.compile(optimizer="rmsprop", loss=crf.loss_function, metrics=[crf.accuracy])
for epoch in list(range(number_of_epochs)):
model.fit(train_tokens, train_y_cat, nb_epoch=1, batch_size=minibatch_size, verbose=True, shuffle=True)
model.predict(dev_tokens, verbose=0)
火车令牌-(452853,7) train_y_cat-(452853,26) 宁7 结节26
请注意,当我使用除crf层之外的lstm(100)时,此模型可以正常工作。