search

wenwenyu / PICK-pytorch

Code for the paper "PICK: Processing Key Information Extraction from Documents using Improved Graph Learning-Convolutional Networks" (ICPR 2020)
https://arxiv.org/abs/2004.07464
MIT License
553 stars 191 forks source link

Add tensorboard log (mEA, mEF, ..) with same steps #116

Open yellowjs0304 opened 2 years ago

yellowjs0304 commented 2 years ago

Hi, First of all Thank you for sharing nice works. It seems really good in NER tasks.

I wanna check with my own dataset, So I customized my data and have trained. In this steps, I have some questions as below.

  1. How can I add tensorboard log with metrics(ex. mEA, mEF, mER, mEP).

    • I added the function self.writer.add_scalar("mEP/steps", val_result_dict['overall']['mEP']) after Step Validation line, because I wanna check the metric results in every steps.
    • But It returns like below, mEA steps is not same with other metric(gl_loss). I don't know the way to get right steps. image

  2. Is there any function to visualize the relations of each entities.(like Key-Values)?

    • I wanna check that model predicted right Key-Value mapping by graph embedding(GCN).. Do you have any functions about it? If not so, could you please give some tips to implement function?

  3. Is there any visualization function to inference the result of models on each of images+bbox+predicted labels like below? image

I'll wait for your reply. Thank you 😊