Hugging face是一个开源的模型社区,里边包含各种各样的模型。Hugging face在github上开源了一个Transformers库,这个库非常的好用,我们可以用这个库做模型的推理、微调。推荐这个库的另一个原因是简单好用,我们看一下huggingface的网站
huggingface除了开源模型外,还提供数据集、模型的结构与训练代码,我们甚至可以无痛在这个框架下做模型finetune。
快速开始
我们使用一个命名实体识别的模型做一下推理,结果见下图:
除了使用网页的API,我们还能使用python代码做推理(需要搭建环境),代码如下:
from transformers import AutoTokenizer, AutoModelForTokenClassification
from transformers import pipeline
tokenizer = AutoTokenizer.from_pretrained("dslim/bert-base-NER")
model = AutoModelForTokenClassification.from_pretrained("dslim/bert-base-NER")
nlp = pipeline("ner", model=model, tokenizer=tokenizer)
example = "My name is Wolfgang and I live in Berlin"
ner_results = nlp(example)
print(ner_results)
结尾
这篇文章的观看情况好的话,将继续更新huggingface的相关内容。