// 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 = "DatasetSplitProto";

message DatasetSplitParams {
  // 0 ~ 100, 剩下的是测试集的大小
  int32 training_data_ratio = 1;
  // 是否需要固定随机数
  // false: sklearn.model_selection.train_test_split(random_state=None)
  // true: sklearn.model_selection.train_test_split(random_state=random_state)
  bool should_fix_random = 2;
  // random_state 不变，则分割出的数据结果不变
  uint64 random_state = 3;
}

message DataSetSplitReport {
    int32 train_set_count = 1;
    int32 test_set_count = 2;
}