ExtractedModel
class ExtractedModel(trees: list[dict], base: float, learning_rate: float, family: str, input_precision: str = 'float64', n_features: int = 0, notes: list[str] = <factory>) -> None
from compileml.compile import ExtractedModel
Float-precision tree ensemble extracted from a source model.
Fields#
| Name | Type | Default | Kind |
|---|---|---|---|
| trees | list[dict] | required | positional |
| base | float | required | positional |
| learning_rate | float | required | positional |
| family | str | required | positional |
| input_precision | str | 'float64' | positional |
| n_features | int | 0 | positional |
| notes | list[str] | <factory> | positional |
Returns#
None