心在飞翔

西安交通大学-心在飞翔队作业一

模型描述

使用模型：qwen-1.8b 数据库：京东情感分类数据库 实验环境：notebook 模型效果： 对情感进行简单的二分类，分为积极和消极两种情绪，根据用户输入的语句判断其内含的情感。

示例代码

import os os.environ['CUDAVISIBLEDEVICES'] = '0' import torch from swift.llm import ( DatasetName, InferArguments, ModelType, SftArguments, infermain, sftmain, appuimain, mergeloramain ) modeltype = ModelType.qwen18b sftargs = SftArguments( modeltype=modeltype, sfttype='lora', traindatasetsample=2000, dataset=[DatasetName.jdsentimentzh], outputdir='output') result = sftmain(sftargs) bestmodelcheckpoint = result['bestmodelcheckpoint'] print(f'bestmodelcheckpoint: {bestmodelcheckpoint}') torch.cuda.emptycache() inferargs = InferArguments( ckptdir=bestmodelcheckpoint, loaddatasetconfig=True, dosample=False) result = infermain(inferargs)

 git clone https://www.modelscope.cn/Virtuosa/cl1.git