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)[source]#
- 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