Agriculture-Classification

This is a model used for classifying whether it is a agricultural question or not.

Usage

In Transformers

from transformers import BertTokenizer, BertForSequenceClassification, pipeline
model_name = "NchuNLP/Agriculture-Classification"
tokenizer = BertTokenizer.from_pretrained(model_name)
model = BertForSequenceClassification.from_pretrained(model_name, num_labels=2)

# Get predictions 
nlp = pipeline('text-classification', model=model, tokenizer=tokenizer)
query = "水稻生長的適宜溫度是多少？"
res = nlp(query)

Authors

Peng-Yi Lin: [email protected]

Yao-Chung Fan: [email protected]

About us

中興大學自然語言處理實驗室研究方向圍繞於深度學習技術在文字資料探勘 (Text Mining) 與自然語言處理 (Natural Language Processing) 方面之研究，目前實驗室成員的研究主題著重於機器閱讀理解 (Machine Reading Comprehension) 以及自然語言生成 (Natural Language Generation) 兩面向。

More Information

For more info about Nchu NLP Lab, visit our Lab Online Demo repo and GitHub.