Traiig procedure

Framework versios

• SWIFT 1.5.1

Base model iformatio

队名：西安交通大学-心在飞翔队 使用模型：qwe-1.8b 数据库：京东情感分类数据库 实验环境：otebook 模型效果： 对情感进行简单的二分类，分为积极和消极两种情绪，根据用户输入的语句判断其内含的情感 示例代码：

使用swift框架

import os os.eviro['CUDAVISIBLEDEVICES'] = '0' import torch from swift.llm import ( DatasetName, IferArgumets, ModelType, SftArgumets, ifermai, sftmai, appuimai, mergeloramai )

选择1.8B模型

modeltype = ModelType.qwe18b sftargs = SftArgumets( modeltype=modeltype,

使用lora方法

sfttype='lora', traidataset_sample=2000,

调用京东的评论数据集

dataset=[DatasetName.jdsetimetzh], outputdir='output') result = sftmai(sft_args)

adapter的⼤⼩: 6.1MiB

bestmodelcheckpoit = result['bestmodelcheckpoit'] prit(f'bestmodelcheckpoit: {bestmodelcheckpoit}') torch.cuda.emptycache() iferargs = IferArgumets( ckptdir=bestmodelcheckpoit, loaddatasetcofig=True, dosample=False) result = ifermai(iferargs)

• BaseModel Class QWeLMHeadModel

Training procedure Framework versions SWIFT 1.5.1 Base model information 队名：西安交通大学-心在飞翔队 使用模型：qwen