ForestInference#
- class cuml.fil.ForestInference(
- *,
- treelite_model=None,
- output_type=None,
- verbose=False,
- is_classifier=False,
- layout='depth_first',
- default_chunk_size=None,
- align_bytes=None,
- precision='single',
- device_id=None,
- ensure_all_finite=False,
- _suppress_deprecation_warning=False,
- Attributes:
- align_bytes
- default_chunk_size
- device_id
- is_classifier
- layout
- precision
- treelite_model
Methods
apply
get_dtype
load
load_from_sklearn
load_from_treelite_model
num_outputs
num_trees
optimize
predict
predict_per_tree
predict_proba