AI21 Labs

自定义模型训练

开发环境

POST

https://api.ai21.com/studio/v1/custom-model

请求示例

Shell

JavaScript

Java

Swift

curl --location --request POST 'https://api.ai21.com/studio/v1/custom-model' \
--header 'Content-Type: application/json' \
--data-raw '{
  "model_type": "j2-mid",
  "num_epochs": 20
}'

响应示例

{}

请求参数

Body 参数application/json

model_type

string

必需

要训练的模型类型。默认值:j2-mid

dataset_id

string

必需

要用来训练模型的数据集。可以通过API请求或Studio中的Datasets标签获取。

model_name

string

必需

训练后的模型名称。必须唯一。

learning_rate

number

可选

0 <= 浮点数 <= 1.0 训练使用的学习率。可选,默认值取决于模型类型。

num_epochs

integer

可选

0 < 整数训练轮次数。可选,默认值=20。

示例

返回响应

🟢200成功

application/json

Body

object {0}

修改于 2023-11-13 02:48:43

自定义模型 API

获取自定义模型列表