Fork me on GitHub
摘要: from torch.optim import Adam from torch.utils.data import DataLoader from dataset.wiki_dataset import BERTDataset from models.bert_model import * impo 阅读全文
posted @ 2020-07-21 13:55 西西嘛呦 阅读(770) 评论(0) 推荐(0)
摘要: from torch.utils.data import Dataset import tqdm import json import torch import random import numpy as np from sklearn.utils import shuffle class BER 阅读全文
posted @ 2020-07-21 13:53 西西嘛呦 阅读(434) 评论(0) 推荐(0)
摘要: 代码出处:https://github.com/aespresso/a_journey_into_math_of_ml/blob/master/04_transformer_tutorial_2nd_part/BERT_tutorial/models/bert_model.py # coding=u 阅读全文
posted @ 2020-07-21 11:45 西西嘛呦 阅读(2097) 评论(0) 推荐(2)
摘要: import os import json # 现在已经有准备好的BERT维基百科训练语料, 已经分割为train_wiki.txt和test_wiki.txt # 语料来源: https://github.com/brightmart/nlp_chinese_corpus # 在准备好的文件里, 阅读全文
posted @ 2020-07-21 11:29 西西嘛呦 阅读(315) 评论(0) 推荐(0)