Treelite
Variables
main_template.h File Reference

template for main function More...

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Variables

const char *const query_functions_definition_template
 
const char *const main_start_template
 
const char *const main_end_multiclass_template
 
const char *const main_end_template
 

Detailed Description

template for main function

Copyright (c) 2018-2020 by Contributors

Author
Hyunsu Cho

Definition in file main_template.h.

Variable Documentation

◆ main_end_multiclass_template

const char* const main_end_multiclass_template
Initial value:
=
R"TREELITETEMPLATE(
for (int i = 0; i < {num_class}; ++i) {{
result[i] = sum[i]{optional_average_field} + ({leaf_output_type})({global_bias});
}}
if (!pred_margin) {{
return pred_transform(result);
}} else {{
return {num_class};
}}
}}
)TREELITETEMPLATE"

Definition at line 58 of file main_template.h.

◆ main_end_template

const char* const main_end_template
Initial value:
=
R"TREELITETEMPLATE(
sum = sum{optional_average_field} + ({leaf_output_type})({global_bias});
if (!pred_margin) {{
return pred_transform(sum);
}} else {{
return sum;
}}
}}
)TREELITETEMPLATE"

Definition at line 71 of file main_template.h.

◆ main_start_template

const char* const main_start_template
Initial value:
=
R"TREELITETEMPLATE(
#include "header.h"
{array_is_categorical};
{query_functions_definition}
{pred_transform_function}
{predict_function_signature} {{
)TREELITETEMPLATE"

Definition at line 46 of file main_template.h.

◆ query_functions_definition_template

const char* const query_functions_definition_template
Initial value:
=
R"TREELITETEMPLATE(
size_t get_num_class(void) {{
return {num_class};
}}
size_t get_num_feature(void) {{
return {num_feature};
}}
const char* get_pred_transform(void) {{
return "{pred_transform}";
}}
float get_sigmoid_alpha(void) {{
return {sigmoid_alpha};
}}
float get_global_bias(void) {{
return {global_bias};
}}
const char* get_threshold_type(void) {{
return "{threshold_type_str}";
}}
const char* get_leaf_output_type(void) {{
return "{leaf_output_type_str}";
}}
)TREELITETEMPLATE"

Definition at line 15 of file main_template.h.