bert sklearn:Google BERT模型的sklearn包装器 源码

scikit-learn包装器对BERT进行微调 一个scikit-learning包装器,可基于端口模型,以执行文本和令牌序列任务。 包括可配置的MLP作为文本和文本对任务的最终分类器/回归器 包括用于NER,PoS和分块任务的令牌序列分类器 包括针对科学和生物医学领域的和训练模型。 在尝试! 安装 需要python> = 3.5和pytorch> = 0.4.1 git clone -b master https://github.com/charles9n/bert-sklearn cd bert-sklearn pip install . 基本操作 model.fit(X,y)即