From a16ce680351952b6359c4fb831956868327dac23 Mon Sep 17 00:00:00 2001 From: William Hicks Date: Thu, 25 Aug 2022 18:59:28 -0400 Subject: [PATCH] Correct docstrings --- python/cuml/fil/fil.pyx | 3 +++ 1 file changed, 3 insertions(+) diff --git a/python/cuml/fil/fil.pyx b/python/cuml/fil/fil.pyx index c7b78fa39d..017b410a52 100644 --- a/python/cuml/fil/fil.pyx +++ b/python/cuml/fil/fil.pyx @@ -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 @@ -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 @@ -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