赞
踩
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的相关内容。
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。