Tapex Large Finetuned Sqa

🚀 TAPEX-large微調SQA模型

TAPEX-large模型在SQA數據集上進行了微調。該模型由Qian Liu、Bei Chen、Jiaqi Guo、Morteza Ziyadi、Zeqi Lin、Weizhu Chen、Jian-Guang Lou提出於論文 TAPEX: Table Pre-training via Learning a Neural SQL Executor。原始代碼庫可在 此處 找到。

🚀 快速開始

若要加載該模型並進行推理，可按以下步驟操作：

💻 使用示例

基礎用法

from transformers import BartTokenizer, BartForConditionalGeneration
import pandas as pd

tokenizer = BartTokenizer.from_pretrained("nielsr/tapex-large-finetuned-sqa")
model = BartForConditionalGeneration.from_pretrained("nielsr/tapex-large-finetuned-sqa")

# create table
data = {'Actors': ["Brad Pitt", "Leonardo Di Caprio", "George Clooney"], 'Number of movies': ["87", "53", "69"]}
table = pd.DataFrame.from_dict(data)

# turn into dict
table_dict = {"header": list(table.columns), "rows": [list(row.values) for i,row in table.iterrows()]}

# turn into format TAPEX expects
# define the linearizer based on this code: https://github.com/microsoft/Table-Pretraining/blob/main/tapex/processor/table_linearize.py
linearizer = IndexedRowTableLinearize()
linear_table = linearizer.process_table(table_dict)

# add question
question = "how many movies does George Clooney have?"
joint_input = question + " " + linear_table

# encode 
encoding = tokenizer(joint_input, return_tensors="pt")

# forward pass
outputs = model.generate(**encoding)

# decode
tokenizer.batch_decode(outputs, skip_special_tokens=True)

📄 許可證

本項目採用Apache-2.0許可證。