Module hummingbird.ml.operator_converters.constants
Constants used in the Hummingbird converters are defined here.
Expand source code
# -------------------------------------------------------------------------
# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License. See License.txt in the project root for
# license information.
# --------------------------------------------------------------------------
"""
Constants used in the Hummingbird converters are defined here.
"""
BASE_PREDICTION = "base_prediction"
"""Alpha"""
LEARNING_RATE = "learning_rate"
"""Learning Rate"""
GET_PARAMETERS_FOR_TREE_TRAVERSAL = "get_parameters_for_tree_trav"
"""Which function to use to extract the parameters for the tree traversal strategy"""
REORDER_TREES = "reorder_trees"
"""Whether to reorder trees in multiclass tasks"""
ONNX_INPUTS = "onnx_inputs"
"""The inputs to the onnx model"""
ONNX_TEST_INPUT = "onnx_test_input"
"""The test input data for the onnx model"""
OFFSET = "offset_"
"""offset of the sklearn anomaly detection implementation"""
MAX_SAMPLES = "max_samples_"
"""max_samples of sklearn isolation forest implementation"""
Global variables
var BASE_PREDICTION
-
Alpha
var GET_PARAMETERS_FOR_TREE_TRAVERSAL
-
Which function to use to extract the parameters for the tree traversal strategy
var LEARNING_RATE
-
Learning Rate
var MAX_SAMPLES
-
max_samples of sklearn isolation forest implementation
var OFFSET
-
offset of the sklearn anomaly detection implementation
var ONNX_INPUTS
-
The inputs to the onnx model
var ONNX_TEST_INPUT
-
The test input data for the onnx model
var REORDER_TREES
-
Whether to reorder trees in multiclass tasks