FateWithShellmiaoComment/Fate/python/federatedml/protobuf/proto/boosting-tree-model-meta.proto

syntax = "proto3";

package com.webank.ai.fate.core.mlmodel.buffer;
option java_outer_classname = "BoostTreeModelMetaProto";

message ObjectiveMeta {
    string objective = 1;
    repeated double param = 2;
}

message CriterionMeta {
    string criterion_method = 1;
    repeated double criterion_param = 2;
}

message DecisionTreeModelMeta {
    CriterionMeta criterion_meta = 1;
    int32 max_depth = 2;
    int32 min_sample_split = 3;
    double min_impurity_split = 4;
    int32 min_leaf_node = 5;
    bool use_missing = 6;
    bool zero_as_missing = 7;
}

message QuantileMeta {
    string quantile_method = 1;
    int32 bin_num = 2;
}

message BoostingTreeModelMeta {
    DecisionTreeModelMeta tree_meta = 1;
    double learning_rate = 2;
    int32 num_trees = 3;
    QuantileMeta quantile_meta = 4;
    ObjectiveMeta objective_meta = 5;
    string task_type = 6;
    bool n_iter_no_change = 7;
    double tol = 8;
    bool use_missing = 9;
    bool zero_as_missing = 10;
    string work_mode = 11;
    string module = 12;
    string boosting_strategy = 13;
}

message TransformerMeta{
    BoostingTreeModelMeta tree_meta=1;
    string model_name=2;
}