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
Transformation functions are introduced. Before this step `leaves` was able to output only raw predictions. Here is new bool option named `loadTransformation` adedd to all model load functions: `XGEnsembleFromReader`, `XGEnsembleFromFile`, `XGBLinearFromReader`, `XGBLinearFromFile`, `SKEnsembleFromReader`, `SKEnsembleFromFile`, `LGEnsembleFromJSON`, `LGEnsembleFromReader`, `LGEnsembleFromFile`.
Also, `NClasses``Ensemble` method will be renamed to `NRawOutputGroups` while keeping the same meaning - number of values that model provides for every object in raw predictions. There is also added `NOutputGroups` - number of values that model provides for every object after applying transformation function. Generally, that means that transformation function can change outputs dimensionality. Please note, if current transformation funciton is `raw`:
@@ -69,6 +71,8 @@ In order to use XGBoost model, just change `leaves.LGEnsembleFromFile`, to `leav
69
71
70
72
Documentation is hosted on godoc ([link](https://godoc.org/github.com/dmitryikh/leaves)). Documentation contains complex usage examples and full API reference. Some additional information about usage examples can be found in [leaves_test.go](leaves_test.go).
71
73
74
+
Some additional information on new features and backward compatibility can be found in [NOTES.md](NOTES.md).
75
+
72
76
## Benchmark
73
77
74
78
Below are comparisons of prediction speed on batches (~1000 objects in 1 API
@@ -104,9 +108,9 @@ Single thread:
104
108
## Limitations
105
109
106
110
* LightGBM models:
107
-
*no support transformations functions (sigmoid, lambdarank, etc). Output scores is _raw scores_
111
+
*limited support of transformation functions (support only sigmoid, softmax)
108
112
* XGBoost models:
109
-
*no support transformations functions. Output scores is _raw scores_
113
+
*limited support of transformation functions (support only sigmoid, softmax)
110
114
* could be slight divergence between C API predictions vs. _leaves_ because of floating point convertions and comparisons tolerances
111
115
* scikit-learn tree models:
112
116
* no support transformations functions. Output scores is _raw scores_ (as from `GradientBoostingClassifier.decision_function`)
0 commit comments