QuickRank  v2.0
QuickRank: A C++ suite of Learning to Rank algorithms
Public Member Functions | Private Member Functions | List of all members
quickrank::io::GenOblivious Class Reference

This class is a code generator on QuickRank XML files. More...

#include <generate_oblivious.h>

Public Member Functions

void generate_oblivious_code (const std::string, const std::string)
 Generates the C++ implementation of the model scoring function. More...
 

Private Member Functions

void model_tree_get_leaves (pugi::xml_node &, std::vector< std::string > &)
 
void model_tree_get_feature_ids (pugi::xml_node &, std::vector< unsigned int > &)
 
void model_tree_get_thresholds (pugi::xml_node &, std::vector< std::string > &)
 

Detailed Description

This class is a code generator on QuickRank XML files.

The XML format is used for loading and storing ranking models.

Member Function Documentation

void quickrank::io::GenOblivious::generate_oblivious_code ( const std::string  model_filename,
const std::string  code_filename 
)

Generates the C++ implementation of the model scoring function.

This applies to forests of oblivious trees.

Parameters
model_filenamePreviously saved XML ranker model.
code_filenameOutput source code file name.
void quickrank::io::GenOblivious::model_tree_get_feature_ids ( pugi::xml_node &  split_xml,
std::vector< unsigned int > &  features 
)
private
void quickrank::io::GenOblivious::model_tree_get_leaves ( pugi::xml_node &  split_xml,
std::vector< std::string > &  leaves 
)
private
void quickrank::io::GenOblivious::model_tree_get_thresholds ( pugi::xml_node &  split_xml,
std::vector< std::string > &  thresholds 
)
private

The documentation for this class was generated from the following files: