Skip to content

Commit

Permalink
Correct docstrings
Browse files Browse the repository at this point in the history
  • Loading branch information
wphicks committed Aug 25, 2022
1 parent dfc3c48 commit a16ce68
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions python/cuml/fil/fil.pyx
Original file line number Diff line number Diff line change
Expand Up @@ -614,6 +614,7 @@ class ForestInference(Base,
precision : string (default='native')
precision of weights and thresholds of the FIL model loaded from
the treelite model.
- ``'native'``: load in float64 if the treelite model contains float64
weights or thresholds, otherwise load in float32
- ``'float32'``: always load in float32, may lead to loss of precision
Expand Down Expand Up @@ -801,6 +802,7 @@ class ForestInference(Base,
precision : string (default='native')
precision of weights and thresholds of the FIL model loaded from
the treelite model.
- ``'native'``: load in float64 if the treelite model contains
float64 weights or thresholds, otherwise load in float32
- ``'float32'``: always load in float32, may lead to loss of
Expand Down Expand Up @@ -893,6 +895,7 @@ class ForestInference(Base,
precision : string (default='native')
precision of weights and thresholds of the FIL model loaded from
the treelite model.
- ``'native'``: load in float64 if the treelite model contains
float64 weights or thresholds, otherwise load in float32
- ``'float32'``: always load in float32, may lead to loss of
Expand Down

0 comments on commit a16ce68

Please sign in to comment.