google-research / tapas

End-to-end neural table-text understanding models.
Apache License 2.0
1.15k stars 217 forks source link

Run on custom data #7

Closed PunitShah1988 closed 4 years ago

PunitShah1988 commented 4 years ago

Hi,

I am curious if you can guide or confirm how I can use TAPAS framework on my custom data. I couldn't find any resources which talks about this.

Thank you.

eisenjulian commented 4 years ago

Hello @PunitShah1988, this is the current data preparation pipeline:

The colab we added to the repo shows how to create an Interaction, and save it directly as a tf_record. If you do that for all of your data then you can run the train job directly.

Alternatively you can try to get your data into the same format as SQA TSVs or add your own conversion code to task_utils. You may also want to think about which of the three datasets your problem is more similar to, is it conversational (SQA), does it require aggregations (WTQ, WikiSQL), do you have supervision for which cells to aggregate (WIkiSQL), etc...

Hope this helps, good luck!