README.md 1.87 KB
Newer Older
yanke_chen's avatar
yanke_chen committed
1
# DataAnalysis
sharkgene@qq.com's avatar
sharkgene@qq.com committed
2
将大模型推理测试生成的excel数据文件,按照特定方式生成柱状图进行比较
yanke_chen's avatar
yanke_chen committed
3

sharkgene@qq.com's avatar
sharkgene@qq.com committed
4
配置文件data_config.json示例说明:
yanke_chen's avatar
yanke_chen committed
5
```
sharkgene@qq.com's avatar
sharkgene@qq.com committed
6
7
8
9
10
11
{
  "filter": {
    "并发数": [],
    "模型": []

  },
12
13
14
15
  "dist_cols": ["模型", "卡数"],
  "group_cols": [["并发数"], ["输入长度(tokens)", "输出长度(tokens)"]],
  "key_cols": ["卡类型", "vLLM版本", "V0/V1 Engine"],
  "metric_cols": ["平均首字延时TTFT(ms)", "平均生成时间TPOT(ms)", "生成吞吐量(tokens/s)", "总吞吐量(tokens/s)"],
sharkgene@qq.com's avatar
sharkgene@qq.com committed
16
17
  "files": [
    {
18
      "file": "t.xlsx",
sharkgene@qq.com's avatar
sharkgene@qq.com committed
19
20
21
22
23
24
25
26
27
28
      "sheets": [],
      "column_mapping": {
      },
      "column_replace": {
      }
    }
  ]
}

filter: 过滤条件,不设置或者为空则不过滤
29
30
31
32
dist_cols: 选择输出文件的条件,例如["模型", "卡类型", "卡数"]则将相同的模型、卡类型和卡数的所有数据生成以个文件
group_cols: 可以进行最多2重分组,每层一个图,第2层以不同柱状显示数据。如果命令行参数指定合并,则所有合成一个图,分组之间间隔开
key_cols: 分组内的键值
metric_cols: 度量列,不设置缺省上述4个,会将所有非数值格式转换为0
sharkgene@qq.com's avatar
sharkgene@qq.com committed
33
34
35
36
files: 指定文件信息, 可以使用多个文件
    file: 文件名
    sheets: 可以指定处理的sheet,不指定则遍历所有
    column_mapping: 列名重映射,用于处理多个文件时表头信息略有差异的情况
37
38
参考配置说明:

yanke_chen's avatar
yanke_chen committed
39
```
sharkgene@qq.com's avatar
sharkgene@qq.com committed
40
## 使用方法
yanke_chen's avatar
yanke_chen committed
41
```
sharkgene@qq.com's avatar
sharkgene@qq.com committed
42
43
44
45
46
47
48
49
50
51
python3 plot_comparison.py --help
usage: plot_comparison.py [-h] [--配置 配置] [--输出目录 输出目录] [--合并分组]

绘制模型性能对比图表

options:
  -h, --help            show this help message and exit
  --配置 配置, -f 配置        数据配置文件路径
  --输出目录 输出目录, -d 输出目录  输出图表目录
  --合并分组, -m            将第一层分组合并到一张图中
yanke_chen's avatar
yanke_chen committed
52
```
sharkgene@qq.com's avatar
sharkgene@qq.com committed
53
##