Adjust operation names to namefinder.py
diff --git a/tf-ner-poc/src/main/java/org/apache/opennlp/tf/guillaumegenthial/SequenceTagging.java b/tf-ner-poc/src/main/java/org/apache/opennlp/tf/guillaumegenthial/SequenceTagging.java
index 653b58a..637ccec 100644
--- a/tf-ner-poc/src/main/java/org/apache/opennlp/tf/guillaumegenthial/SequenceTagging.java
+++ b/tf-ner-poc/src/main/java/org/apache/opennlp/tf/guillaumegenthial/SequenceTagging.java
@@ -61,15 +61,15 @@
FeedDictionary fd = FeedDictionary.create(tokenIds);
List<Tensor<?>> run = session.runner()
- .feed("char_ids:0", fd.getCharIdsTensor())
- .feed("dropout:0", fd.getDropoutTensor())
- .feed("sequence_lengths:0", fd.getSentenceLengthsTensor())
- .feed("word_ids:0", fd.getWordIdsTensor())
- .feed("word_lengths:0", fd.getWordLengthsTensor())
- .fetch("proj/logits", 0)
+ .feed("chars/char_ids:0", fd.getCharIdsTensor())
+ // TODO: missing in the python code ...
+ //.feed("dropout:0", fd.getDropoutTensor())
+ .feed("words/sequence_lengths:0", fd.getSentenceLengthsTensor())
+ .feed("words/word_ids:0", fd.getWordIdsTensor())
+ .feed("chars/word_lengths:0", fd.getWordLengthsTensor())
+ .fetch("logits", 0)
.fetch("trans_params", 0).run();
-
float[][][] logits = new float[fd.getNumberOfSentences()][fd.getMaxSentenceLength()][indexTagger.getNumberOfTags()];
run.get(0).copyTo(logits);
