bert-component/config.json at master · YC-wind/bert-component · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
{
  "params": {
    "vocab_file": {
      "title": "bert词汇表",
      "desc": "指定bert所需的词汇表，中文/英文",
      "default": "./bert/vocab.txt",
      "type": "string"
    },
    "csv_file": {
      "title": "训练文件",
      "desc": "指定所需训练文件",
      "default": "",
      "type": "string"
    },
    "column_name_x1": {
      "title": "列名x1",
      "desc": "输入文本片段一",
      "default": "x1",
      "type": "string"
    },
    "column_name_x2": {
      "title": "列名x2",
      "desc": "输入文本片段二，默认为空表示不启用",
      "default": "",
      "type": "string"
    },
    "column_name_y": {
      "title": "列名y",
      "desc": "模型需要预测的标签",
      "default": "y",
      "type": "string"
    }
  },
  "paramExample": {
    "input": {
      "t1": "/tmp/"
    },
    "output": {
      "r1": "/tmp/"
    },
    "column_name_x1": "x1",
    "column_name_x2": "",
    "column_name_y": "y"
  },
  "executionParam": {
    "entry": "sklearn-xgboost-train-component/main.py"
  }
}