有任何问题欢迎在下面留言
本篇文章的代码运行界面均在Jupyter Notebook中进行
本篇文章配套的代码资源已经上传
数据下载链接,也可以通过代码中加载
这个数据集也是Hugging Face内置的数据集,这个数据集有多个语言版本,我们选中文的就好:
这个数据中我们主要关注两个字段:(review_body和review_title),其他的我们暂时不需要用,这个数据中body就是我们的输入,title就是我们的标签输出
from datasets import load_dataset
chinese_dataset = load_dataset("A:/amazon_reviews_multi")
chinese_dataset
DatasetDict({
?train: Dataset({
??features: [‘id’, ‘text’, ‘label’, ‘label_text’],
??num_rows: 200000
?})
?validation: Dataset({
??features: [‘id’, ‘text’, ‘label’, ‘label_text’],
??num_rows: 5000
?})
?test: Dataset({
??features: [‘id’, ‘text’, ‘label’, ‘label_text’],
??num_rows: 5000
?})
})
这个数据集中,数据是比较大的,因为包含了各种各样的数据,训练集达到 了20万,验证测试都是5千
def show_samples(dataset, num_samples=3, seed=40):
sample = dataset["train"].shuffle(seed=seed).select(range(num_samples))
for example in sample:
print(f"\n'>> Title: {example['review_title']}'")
print(f"'>> Review: {example['review_body']}'")
show_samples(chinese_dataset)
‘>> Title: 重修版的结局’ ’
‘>> Review:’
重修版的结局还是跟原版没什么很大出入…虽然说把原来伏笔用上句式古风可是结局感觉给人有点仓促’
‘>> Title: 盗版书!!!’
‘>> Review:’
这是盗版书,中间翻不开,胶装在一起的,如果想要翻开看全图,书都要撕掉的感觉,装订特别烂,我才翻了几次就有掉页的趋势,后悔后悔’
‘>> Title: 一分钱一分货’ ‘>> Review: 除了便宜真没什么好的,即便4档面包靠里一面还是会胡,设7档有什么意义。’
在原始的数据中,有很多种类