You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
gbdt-rs is a gradient boost decision tree library written in Safe Rust. There is no unsafe rust code in the library.
6
+
MesaTEE GBDT-RS is a gradient boost decision tree library written in Safe Rust. There is no unsafe rust code in the library.
7
7
8
-
gbdt-rs provides the training and inference capabilities. And it can use the models trained by [xgboost](https://xgboost.readthedocs.io/en/latest/) to do inference tasks.
8
+
MesaTEE GBDT-RS provides the training and inference capabilities. And it can use the models trained by [xgboost](https://xgboost.readthedocs.io/en/latest/) to do inference tasks.
9
9
10
-
New! The gbdt-rs[paper](gbdt.pdf) has been [accepted by IEEE S&P'19](https://www.ieee-security.org/TC/SP2019/program-posters.html)!
10
+
New! The MesaTEE GBDT-RS[paper](gbdt.pdf) has been [accepted by IEEE S&P'19](https://www.ieee-security.org/TC/SP2019/program-posters.html)!
11
11
12
12
13
13
# Supported Task
14
14
## Supppoted task for both training and inference
15
15
1. Linear regression: use SquaredError and LAD loss types
16
16
2. Binary classification (labeled with 1 and -1): use LogLikelyhood loss type
17
17
## Compatibility with xgboost
18
-
At this time, gbdt-rs support to use model trained by xgboost to do inference. The model should be trained by xgboost with following configruation:
18
+
At this time, MesaTEE GBDT-RS support to use model trained by xgboost to do inference. The model should be trained by xgboost with following configruation:
19
19
20
20
1. booster: gbtree
21
21
2. objective: "reg:linear", "reg:logistic", "binary:logistic", "binary:logitraw", "multi:softprob", "multi:softmax" or "rank:pairwise".
22
22
23
-
We have tested that gbdt-rs is compatible with xgboost 0.81 and 0.82
23
+
We have tested that MesaTEE GBDT-RS is compatible with xgboost 0.81 and 0.82
24
24
25
25
# Quick Start
26
26
## Training Steps
@@ -84,7 +84,7 @@ We have tested that gbdt-rs is compatible with xgboost 0.81 and 0.82
84
84
* Note convert_xgboost.py depends on xgboost python libraries. The converted model can be used on machines without xgboost
85
85
3. In rust code, call GBDT::load_from_xgboost(model_path, objective) to load the model
86
86
4. Do inference
87
-
5. (optional) Call GBDT::save_model to save the model to gbdt-rs native format.
87
+
5. (optional) Call GBDT::save_model to save the model to MesaTEE GBDT-RS native format.
88
88
89
89
## Example code
90
90
* "reg:linear": examples/test-xgb-reg-linear.rs
@@ -97,12 +97,12 @@ We have tested that gbdt-rs is compatible with xgboost 0.81 and 0.82
97
97
98
98
# Multi-threading
99
99
## Training:
100
-
At this time, training in gbdt-rs is single-threaded.
100
+
At this time, training in MesaTEE GBDT-RS is single-threaded.
101
101
## Inference:
102
102
The related inference functions are single-threaded. But they are thread-safe. We provide an inference example using multi threads in example/test-multithreads.rs
103
103
104
104
# SGX usage
105
-
Because gbdt-rs is written in pure rust, with the help of [rust-sgx-sdk](https://github.com/baidu/rust-sgx-sdk), it can be used in sgx enclave easily as:
105
+
Because MesaTEE GBDT-RS is written in pure rust, with the help of [rust-sgx-sdk](https://github.com/baidu/rust-sgx-sdk), it can be used in sgx enclave easily as:
0 commit comments