diff options
Diffstat (limited to 'Fundamentals_of_Accelerated_Data_Science/3-07_triton.ipynb')
| -rw-r--r-- | Fundamentals_of_Accelerated_Data_Science/3-07_triton.ipynb | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/Fundamentals_of_Accelerated_Data_Science/3-07_triton.ipynb b/Fundamentals_of_Accelerated_Data_Science/3-07_triton.ipynb index 757b3fa..47d586c 100644 --- a/Fundamentals_of_Accelerated_Data_Science/3-07_triton.ipynb +++ b/Fundamentals_of_Accelerated_Data_Science/3-07_triton.ipynb @@ -120,11 +120,14 @@ }, { "cell_type": "code", - "execution_count": 3, + "execution_count": null, "id": "61d898fb-a8d2-4d1c-a13f-2c4be6c18969", "metadata": {}, "outputs": [], "source": [ + "# RAPIDS Forest Inference Library\n", + "# FIL returns class labels or probabilities\n", + "# mem / disk\n", "config_text = f\"\"\"backend: \"fil\"\n", "max_batch_size: 32768\n", "input [ \n", @@ -148,7 +151,7 @@ " value: {{ string_value: \"xgboost_json\" }}\n", " }},\n", " {{\n", - " key: \"output_class\"\n", + " key: \"output_class\" \n", " value: {{ string_value: \"false\" }}\n", " }},\n", " {{\n", |