This is an automated email from the ASF dual-hosted git repository.
joern pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/opennlp-sandbox.git
The following commit(s) were added to refs/heads/master by this push:
new 8e524dc Fix calculation of namecat accuracy
8e524dc is described below
commit 8e524dce9069ba47005b78fd2b70691a699a67f8
Author: Jörn Kottmann <[email protected]>
AuthorDate: Fri Feb 15 16:24:27 2019 +0100
Fix calculation of namecat accuracy
---
tf-ner-poc/src/main/python/namecat/namecat.py | 6 ++++--
1 file changed, 4 insertions(+), 2 deletions(-)
diff --git a/tf-ner-poc/src/main/python/namecat/namecat.py
b/tf-ner-poc/src/main/python/namecat/namecat.py
index 2a6e23f..7e3abf3 100644
--- a/tf-ner-poc/src/main/python/namecat/namecat.py
+++ b/tf-ner-poc/src/main/python/namecat/namecat.py
@@ -177,7 +177,8 @@ def main():
feed_dict = {dropout_keep_prob: 0.5, char_ids_ph:
name_train_batch, name_lengths_ph: name_train_length, y_ph: label_train_batch}
_, probs = sess.run([train_op, probs_op], feed_dict)
- acc_train.append((batch_size -
np.sum(np.abs(label_train_batch - np.argmax(probs, axis=1)))) / batch_size)
+ acc_train.append((batch_size -
np.sum(np.minimum(np.abs(label_train_batch - np.argmax(probs, axis=1)),
+
np.full((batch_size), 1)))) / batch_size)
print("Train acc: " + str(np.mean(acc_train)))
@@ -189,7 +190,8 @@ def main():
feed_dict = {dropout_keep_prob: 1, char_ids_ph:
name_dev_batch, name_lengths_ph: name_dev_length, y_ph: label_dev_batch}
probs = sess.run(probs_op, feed_dict)
- acc_dev.append((batch_size - np.sum(np.abs(label_dev_batch -
np.argmax(probs, axis=1)))) / batch_size)
+ acc_dev.append((batch_size -
np.sum(np.minimum(np.abs(label_dev_batch - np.argmax(probs, axis=1)),
+
np.full((batch_size), 1)))) / batch_size)
print("Dev acc: " + str(np.mean(acc_dev)))