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)))
 

Reply via email to