// Copyright 2023 Ant Group Co., Ltd.
// 
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
// 
//   http://www.apache.org/licenses/LICENSE-2.0
// 
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
//

syntax = "proto3";

package secretflowapis.v1.sdc.teeapps.params;

option java_package = "com.alipay.secretflow.secretflowapis.v1.sdc.teeapps.params";
option java_outer_classname = "PredictProto";

message PredictionParams {
  message OutputControl {
    // 是否将 输入数据集 的label列输出到 预测分数文件
    // 如果是，预测发起方 的 输入数据集 必须包含label列
    bool output_label = 1;
    // 输出到 预测分数文件 时，label列使用的名称
    string label_field_name = 2;
    // 输出到 预测分数文件 时，预测打分使用的名称
    string score_field_name = 3;
    // 是否将输入数据集的id列输出到 预测分数文件
    // 如果是，预测发起方 的 输入数据集 必须包含label列
    bool output_id = 4;
    // 输出到 预测分数文件 时，id列使用的名称
    string id_field_name = 5;
    // 将 输入数据集 的[col_names]列，输出到 预测分数文件
    // 如果是，预测发起方 的 输入数据集 必须包含[col_names]列
    repeated string col_names = 6;
  }
  // 输出选项
  OutputControl output_control = 1;
}