当前位置:   article > 正文

Hugging Face: 拥开源模型社区_huggingfac

huggingfac

        Hugging face是一个开源的模型社区,里边包含各种各样的模型。Hugging face在github上开源了一个Transformers库,这个库非常的好用,我们可以用这个库做模型的推理、微调。推荐这个库的另一个原因是简单好用,我们看一下huggingface的网站

        huggingface除了开源模型外,还提供数据集、模型的结构与训练代码,我们甚至可以无痛在这个框架下做模型finetune

快速开始

        我们使用一个命名实体识别的模型做一下推理,结果见下图:

        除了使用网页的API,我们还能使用python代码做推理(需要搭建环境),代码如下:

  1. from transformers import AutoTokenizer, AutoModelForTokenClassification
  2. from transformers import pipeline
  3. tokenizer = AutoTokenizer.from_pretrained("dslim/bert-base-NER")
  4. model = AutoModelForTokenClassification.from_pretrained("dslim/bert-base-NER")
  5. nlp = pipeline("ner", model=model, tokenizer=tokenizer)
  6. example = "My name is Wolfgang and I live in Berlin"
  7. ner_results = nlp(example)
  8. print(ner_results)
结尾

        这篇文章的观看情况好的话,将继续更新huggingface的相关内容。

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/小蓝xlanll/article/detail/281318
推荐阅读
相关标签
  

闽ICP备14008679号