深度学习,NLP,分类,文本,bert,distilbert,albert,xlnet,罗伯塔,gpt2,torch,pytorch,主动学习,增强,数据
Manteia的Python项目详细描述
曼提亚-宣布好消息
为自然语言处理设计神经网络。深度学习在自然语言处理(NLP)中得到了广泛的应用,因为 它非常适合学习复杂的句子结构和各种单词的语义接近性。 数据清理,构建模型(Bert,Roberta,Distilbert,XLNet,Albert,GPT,GPT2), 质量测量培训,并最终在多个数据集(20newsgroups、SST-2、PubMed_20k_RCT、DBPedia、Amazon Review Full、Amazon Review Polarity)上可视化结果。在
您可以使用pip安装:
pip install Manteia
对于与GPU和cuda一起使用,我们建议使用Anaconda:
conda create-n manteia_env python=3.7
conda activate manteia_env
conda install pytorch
pip install manteia
使用分类示例:
from Manteia.Classification import Classification
from Manteia.Model import Model
documents = ['What should you do before criticizing Pac-Man? WAKA WAKA WAKA mile in his shoe.','What did Arnold Schwarzenegger say at the abortion clinic? Hasta last vista, baby.']
labels = ['funny','not funny']
model = Model(model_name ='roberta')
cl=Classification(model,documents,labels,process_classif=True)
使用生成示例:
^{pr2}$麻省理工学院是根据这一准则授权的。在
- 项目
标签: