How to run prediction on new PDF document

[Sharathmk99]

Hi,

Thank you for your amazing paper and source code.
I was able to successfully run your code.
Based on my understanding, input will be table cell bounding box, table position(row, col) and y will be its row or cell. Is my understanding correct?

If i have new PDF document which contains only Table(Assume i have cropped table from document). How can use GFTE to get rows and col of the table?

Thank you!!

[cccusername]

@Sharathmk99 hello, I run this code but i meet some trouble as follow:
TypeError: knn_graph() takes from 2 to 6 positional arguments but 7 were given
could you tell me which vision of pytorch and torch_geometric you use, thank you!

[jingmouren]

pyg-team/pytorch_geometric#1521 doesn't solve.
@Sharathmk99. what's your torch_geometric and torch_cluster version?

[BrowningWan]

@cccusername Just delete the arguments "num_workers" in line 129 from torch_geometric.nn.pool.init.py.

[Sharathmk99]

@jingmouren Both are latest versions

[harirajeev]

@Sharathmk99 , where you able to run inference on the code , for a new pdf document ?

[Sharathmk99]

@harirajeev I didn't try again. Because i think we also need one more prediction if two words belongs to same cells.

I started with TIES-2.0 repository, but its based on tensorflow. I'll keep you updated if i was able to solve.

[Sharathmk99]

In meantime if anyone solves the problem, please update here.