# cmcc-ptuning

**Repository Path**: gumbou/cmcc-ptuning

## Basic Information

- **Project Name**: cmcc-ptuning
- **Description**: cmcc数据集进行glm2-6b微调
- **Primary Language**: Unknown
- **License**: Not specified
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 0
- **Forks**: 0
- **Created**: 2023-10-11
- **Last Updated**: 2023-10-23

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# cmcc-kp0.1数据集进行p-tuning微调

## 各文件解释

- dataset
  - kp0.1
  - seed=1
- api_ptune.py
  - 定义glm提问的api格式
  - 启动glm模型服务
- preprocess.py
  - 对数据集进行预处理，因为p-tuning需要json格式
- predict.py
  - 对微调后的大模型进行预测
  - 然后保存预测csv文件
  - 计算准确率
- ptuning
  - 微调的工程文件
  - 在train.sh里配置
    - train_file *训练集路径*
    - prompt_coloum  *prompt的列名*
    - response_coloum *label的列名*
    - model_name_or_path *模型的路径*
    - output_dir *输出的路径*