文本分类

文本分类模型

代码示例

1https://api-inference.huggingface.co/models/{model}

1// 接口地址
2const endpointUrl = "https://api-inference.huggingface.co/models/distilbert-base-uncased-finetuned-sst-2-english";
3
4const headerParameters = {
5  "Authorization": "Bearer {{ACCESS_TOKEN}}"
6};
7// 待分类的输入文本
8const data = JSON.stringify({
9  inputs: "Call me Ishmael. Some years ago—never \
10              mind how long precisely—having little or no money in my purse, \
11              and nothing particular to interest me on shore, I thought I would \
12              sail about a little and see the watery part of the world. It is a \
13              way I have of driving off the spleen, and regulating the circulation.",
14  options: {
15    wait_for_model: true
16  }
17});
18
19const options = {
20  method: "POST",
21  headers: headerParameters,
22  body: data
23};
24
25async function textClassification() {
26  try {
27    const response = await fetch(endpointUrl, options);
28    printResponse(response);
29  } catch (error) {
30    printError(error);
31  }
32}
33
34textClassification();

1inputs: "Dickens is one of those writers who are \
2         well worth stealing. Even the burial of his body in Westminster \
3         Abbey was a species of theft, if you come to think of it"

1inputs: ["To be, or not to be",
2        "With mirth and laughter let old wrinkles come"]

零样本分类模型

代码示例

1https://api-inference.huggingface.co/models/{model}

1// 接口地址
2const endpointUrl = "https://api-inference.huggingface.co/models/facebook/bart-large-mnli";
3
4const headerParameters = {
5  "Authorization": "Bearer {{ACCESS_TOKEN}}"
6};
7// 待分类的输入文本
8const data = JSON.stringify({
9  inputs: "Machine Learning skills are some of the most sought-after in the modern \
10  job market. Modern ML Engineers make dozens of thousands of dollars more per year \
11  than other developers. ",
12  parameters: {
13    candidate_labels: ["Software Engineer", "Machine Learning Engineer", "Electrical Engineer"]
14  },
15  options: {
16    wait_for_model: true
17  }
18});
19
20const options = {
21  method: "POST",
22  headers: headerParameters,
23  body: data
24};
25
26async function textClassification() {
27  try {
28    const response = await fetch(endpointUrl, options);
29    printResponse(response);
30  } catch (error) {
31    printError(error);
32  }
33}
34
35textClassification();