Welcome to OGeek Q&A Community for programmer and developer-Open, Learning and Share
Welcome To Ask or Share your Answers For Others

Categories

0 votes
585 views
in Technique[技术] by (71.8m points)

huggingface transformers - What is the multi-label limit for BERT based Hugging Face API?

I'm totally new to the Hugging Face API and I'm trying to figure out the number of labels I can pass in a single API call. The reason I ask this is I tried using 5 labels some time back and it was giving me an error, but now when I tried with 40 labels it worked perfectly fine. So is there a limit at all for the number of labels in the multi-label approach?

The code is as follows-

 payload = json.dumps(
    {"inputs": corpus
    ,"parameters": {"candidate_labels": keywords, "multi_class": True}
    })

headers = {
    'Authorization': ':)',
    'Content-Type': 'application/json'
}
api_response = requests.request("POST", API_URL, headers=headers, data=payload)

Thank You!


与恶龙缠斗过久,自身亦成为恶龙;凝视深渊过久,深渊将回以凝视…
Welcome To Ask or Share your Answers For Others

1 Reply

0 votes
by (71.8m points)
等待大神答复

与恶龙缠斗过久,自身亦成为恶龙;凝视深渊过久,深渊将回以凝视…
OGeek|极客中国-欢迎来到极客的世界,一个免费开放的程序员编程交流平台!开放,进步,分享!让技术改变生活,让极客改变未来! Welcome to OGeek Q&A Community for programmer and developer-Open, Learning and Share
Click Here to Ask a Question

...