跳到主要内容

模型卡片

什么是模型卡片

模型卡片(Model Card)是模型的附带文件,是一个带有元数据的Markdown文件,该文件头部的YAML部分包含模型的元数据。模型卡片涵盖了关键的模型信息,主要通过渲染模型文件中的README.md获得,并最终展示在模型介绍页面。其有助于帮助用户理解和正确使用模型,建议您根据规范撰写模型卡片。

模型卡片应涵盖哪些信息

模型卡片应包含如下内容:

  • 模型名称
  • 模型概述: 包括模型特征和结构
  • 使用方式: 尽可能提供详细范例和代码说明模型的使用方式,对模型运行环境,如何进行模型推理及调优做出介绍和解释
  • 适用场景: 介绍模型的目标场景、预期用途和潜在的限制
  • 训练数据: 说明使用了哪些数据集和训练参数
  • 训练流程: 模型训练的具体过程
  • 模型评估结果: 展示模型的性能评估结果

模型卡片元数据

模型卡片是由YAML元数据和Markdown文本内容组成。您可以通过编辑README.md文件的头部YAML信息来添加元数据,用“---”进行分隔。Markdown文本展示了模型信息和相关说明。

您可以参考以下模版撰写您的模型卡片。

---
# 许可证
license: apache-2.0

# 用户自定义标签
tags:
- image-classification
- customize tags
---

<!--- 以上为YAML格式,提供许可证和任务描述--->

<!--- 以下为markdown格式的模型描述--->

# 模型名称

介绍模型的概要信息

## 模型详情

### 模型描述

对模型进行描述,包括开发者,模型的类型,框架,相关的语言,许可证说明等内容。

## 用途

### 使用方法

介绍模型的使用方法

## 风险和局限性

介绍模型存在的风险和局限性

### 建议

给用户的建议

## 如何开始使用模型

介绍如何开始使用模型

## 训练详情

### 训练数据

介绍模型的训练数据

### 训练流程

介绍模型的训练过程

## 评估

### 测试数据、因素和指标

#### 测试数据

介绍使用的测试数据

#### 因素

介绍测试相关的环境及条件

#### 指标

介绍测试模型使用的指标

### 结果

#### 结果概要

介绍模型测试结果

已支持的模型标签

任务(英文)任务(中文)
image-classification单标签图像分类
image-segmentation通用图像分割
ocr通用字符文字识别
skin-retouching人像美肤
image-to-image图生图
video-detection视频检测
video-segmentation视频分割
video-generation视频生成
video-captioning视频到文本生成
face-detection人脸检测
face-recognition人脸识别
face-image-generation人脸生成
image-object-detection通用目标检测
image-super-resolution图像超分辨
action-detection行为检测
semantic-segmentation语义分割
word-segmentation分词
named-entity-recognition命名实体识别
part-of-speech词性标注
document-segmentation文本分割
text-classification文本分类
sentiment-classification情感分类
nli自然语言推理
conversational对话
translation翻译
text-generation文本生成
text-summarization文本摘要
feature-extraction特征抽取
relation-extraction关系抽取
auto-speech-recognition语音识别
text-to-speech语音合成
audio-classification音频分类
text-to-image文本生成图片
visual-question-answering视觉问答
image-text-retrieval图文检索

已支持的行业标签

行业(中文)行业(英文)
汽车Automotive
制造业Manufacturing
能源Energy
通信与电子信息Telecommunications and Electronic Information
交通运输Transportation and Logistics
建筑与房地产Construction and Real Estate
金融服务Financial Services
农业Agriculture
化工Chemical Industry
环保Environmental Protection
医疗与健康Healthcare and Medical Services
教育与培训Education and Training
食品与饮料Food and Beverage
零售与消费品Retail and Consumer Goods
旅游与酒店Tourism and Hospitality
IT信息技术Information Technology (IT)
文化娱乐Culture and Entertainment