hvlgo commited on
Commit
3a4f1f7
·
1 Parent(s): 2b07748

Update config.yaml

Browse files
Files changed (1) hide show
  1. config.yaml +6 -2
config.yaml CHANGED
@@ -2,7 +2,11 @@ configs:
2
  # 必选项
3
  input_shape: [96, 2] # 表示模型接收的数据为96行x2列
4
  output_shape: [96, 2] # 表示模型输出的数据为96行x2列
 
 
 
 
5
 
6
- note: # 可选项 为用户自定义的备注信息
7
- 'model_type': 'dlinear',
8
  'kernel_size': '25'
 
2
  # 必选项
3
  input_shape: [96, 2] # 表示模型接收的数据为96行x2列
4
  output_shape: [96, 2] # 表示模型输出的数据为96行x2列
5
+
6
+ # 可选项 默认为全部float64,列数为shape对应的列数
7
+ input_type: ["float64","int64"] #输入对应的数据类型,需要与输入列数匹配
8
+ output_type: ["text","int64"] #输出对应的数据类型,需要与输出列数匹配
9
 
10
+ attributes: # 可选项 为用户自定义的备注信息
11
+ 'model_type': 'dlinear'
12
  'kernel_size': '25'