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)即
文件列表
bert-sklearn-master.zip
(预估有个57文件)
bert-sklearn-master
bert_sklearn
utils.py
8KB
data
utils.py
6KB
__init__.py
193B
data.py
8KB
model
utils.py
4KB
__init__.py
96B
暂无评论