闲社服务运行正常AI智能体自动化平台
中文
语言 / Language简体中文SelectedEnglish

选择显示语言

sentiment_classification

Marmeladov/sentiment_classification

发布机构Marmeladov

下载访问条件魔搭来源

此文件夹主要包含微调qwen 1 8b模型得到的模型参数。微调后模型能够判断中文评论或语句包含积极情感还是消极情感. 微调数据集数据集参考了StructBERT情感分类: https://modelscope.cn/models/iic/nlp structbert sentiment-classification chinese-tiny/summary

文本处理
模型详情

模型介绍

中文整理

情感分类模型

模型简介

此文件夹主要包含微调qwen 1 8b模型得到的模型参数。微调后模型能够判断中文评论或语句包含积极情感还是消极情感.

微调数据集数据集参考了StructBERT情感分类:

https://modelscope.cn/models/iic/nlp structbert sentiment-classification chinese-tiny/summary

情感分类任务,通常为输入一段句子或一段话,返回该段话正向/负向的情感极性,在用户评价,观点抽取,意图识别中往往起到重要作用。本中文情感分类模型是基于dianping、jd binary两个数据集部分数据训练出来的情感分类模型。其实这上一句话也是从StructBERT那抄的。

Clone with HTTP

实验环境

python>=3.8

pytorch>=2.0.0

transformers==4.32.0

swift==1.5.3

CUDA 11.4

训练方法

+ 数据集:

商品评论情感预测:https://www.modelscope.cn/datasets/DAMO NLP/jd

大众点评评价数据:https://www.modelscope.cn/datasets/DAMO NLP/yf dianping

+ 数据集处理:

+ 微调模型:ModelType.qwen 1 8b:https://modelscope.cn/models/qwen/Qwen-1 8B

+ 微调方法:swift

示例代码

import os

os.environ['CUDA VISIBLE DEVICES'] = '0'

import torch

from swift.llm import (InferArguments, infer main)

torch.cuda.empty cache()

infer args = InferArguments(

ckpt dir='../sentiment classification',

load dataset config=True,

eval human=True,

do sample=False)

result = infer main(infer args)

...

优先采用原作者提供的中文模型卡片;仅有英文说明时由闲社AI翻译整理。重要参数、许可证和商用范围请以原文为准。

你将获得什么

完整仓库方式包含该固定版本中的权重、配置、分词器及说明文件;指定版本方式只下载所选文件或整组分片。多模态模型的投影文件、配置等依赖,请按作者说明配齐。

完整仓库可能包含多个权重格式,因此需要的磁盘空间可能大于单一格式的模型大小。下载文件不等于完成模型部署。

使用前须知

运行环境

运行模型需要的设备、工具与依赖,以原作者说明为准。

授权范围

是否允许商用、修改和分发,请查看模型许可证。

闲社提供资料与镜像下载方法,不代表已运行模型或完成安全审计。下载的代码应先检查,再执行。

返回顶部