Function BuildModel

tests/example_app/example.c:22–66

  }

TreeliteModelHandle BuildModel() {
  TreeliteModelHandle model;
  TreeliteModelBuilderHandle builder;
  char const* model_metadata
      = "{"
        "  \"threshold_type\": \"float32\","
        "  \"leaf_output_type\": \"float32\","
        "  \"metadata\": {"
        "    \"num_feature\": 2,"
        "    \"task_type\": \"kRegressor\","
        "    \"average_tree_output\": false,"
        "    \"num_target\": 1,"
        "    \"num_class\": [1],"
        "    \"leaf_vector_shape\": [1, 1]"
        "  },"
        "  \"tree_annotation\": {"
        "    \"num_tree\": 1,"
        "    \"target_id\": [0],"
        "    \"class_id\": [0]"
        "  },"
        "  \"postprocessor\": {"
        "    \"name\": \"identity\""
        "  },"
        "  \"base_scores\": [0.0]"
        "}";
  safe_treelite(TreeliteGetModelBuilder(model_metadata, &builder));
  safe_treelite(TreeliteModelBuilderStartTree(builder));
  safe_treelite(TreeliteModelBuilderStartNode(builder, 0));
  safe_treelite(TreeliteModelBuilderNumericalTest(builder, 0, 0.0, 0, "<", 1, 2));
  safe_treelite(TreeliteModelBuilderEndNode(builder));
  safe_treelite(TreeliteModelBuilderStartNode(builder, 1));
  safe_treelite(TreeliteModelBuilderLeafScalar(builder, -1.0));
  safe_treelite(TreeliteModelBuilderEndNode(builder));
  safe_treelite(TreeliteModelBuilderStartNode(builder, 2));
  safe_treelite(TreeliteModelBuilderLeafScalar(builder, 1.0));
  safe_treelite(TreeliteModelBuilderEndNode(builder));
  safe_treelite(TreeliteModelBuilderEndTree(builder));

  safe_treelite(TreeliteModelBuilderCommitModel(builder, &model));

  // Clean up
  safe_treelite(TreeliteDeleteModelBuilder(builder));

  return model;
}

int main() {
  TreeliteModelHandle model = BuildModel();