QuickRank  v2.0
QuickRank: A C++ suite of Learning to Rank algorithms
Public Member Functions | Public Attributes | Static Public Attributes | Private Member Functions | Friends | List of all members
quickrank::learning::CustomLTR Class Reference

#include <custom_ltr.h>

Inheritance diagram for quickrank::learning::CustomLTR:
quickrank::learning::LTR_Algorithm

Public Member Functions

 CustomLTR ()
 
 CustomLTR (const pugi::xml_document &model)
 
virtual ~CustomLTR ()
 
virtual std::string name () const
 Returns the name of the ranker. More...
 
virtual void learn (std::shared_ptr< data::Dataset > training_dataset, std::shared_ptr< data::Dataset > validation_dataset, std::shared_ptr< metric::ir::Metric > metric, size_t partial_save, const std::string model_filename)
 Executes the learning process. More...
 
virtual Score score_document (const Feature *d) const
 Returns the score of a given document. More...
 
virtual pugi::xml_document * get_xml_model () const
 Return the xml model representing the current object. More...
 
- Public Member Functions inherited from quickrank::learning::LTR_Algorithm
 LTR_Algorithm ()
 
 LTR_Algorithm (const pugi::xml_document &model)
 Generates a LTR_Algorithm instance from a previously saved XML model. More...
 
virtual ~LTR_Algorithm ()
 
 LTR_Algorithm (const LTR_Algorithm &other)=delete
 Avoid inefficient copy constructor. More...
 
LTR_Algorithmoperator= (const LTR_Algorithm &)=delete
 Avoid inefficient copy assignment. More...
 
virtual void score_dataset (std::shared_ptr< data::Dataset > dataset, Score *scores) const
 Given and input dateset, the current ranker generates scores for each instance and store the in the scores vector. More...
 
virtual std::shared_ptr< std::vector< Score > > partial_scores_document (const Feature *d) const
 Returns the partial score of a given document, tree by tree. More...
 
virtual void save (std::string model_filename, int suffix=-1) const
 Save the current model to the output_file. More...
 
virtual void print_additional_stats (void) const
 Print additional statistics. More...
 
virtual bool update_weights (std::shared_ptr< std::vector< double >> weights)
 Update the weights for the ensemble models (only). More...
 
virtual std::shared_ptr< std::vector< double > > get_weights () const
 Return the weights for the ensemble models (only). More...
 

Public Attributes

const Score FIXED_SCORE = 666.0
 

Static Public Attributes

static const std::string NAME_ = "CUSTOM"
 

Private Member Functions

virtual std::ostream & put (std::ostream &os) const
 Prints the description of Algorithm, including its parameters. More...
 

Friends

std::ostream & operator<< (std::ostream &os, const CustomLTR &a)
 The output stream operator. More...
 

Additional Inherited Members

- Static Public Member Functions inherited from quickrank::learning::LTR_Algorithm
static std::shared_ptr< LTR_Algorithmload_model_from_file (std::string model_filename)
 Load a model from a given XML file. More...
 

Constructor & Destructor Documentation

quickrank::learning::CustomLTR::CustomLTR ( )
quickrank::learning::CustomLTR::CustomLTR ( const pugi::xml_document &  model)
inline
quickrank::learning::CustomLTR::~CustomLTR ( )
virtual

Member Function Documentation

pugi::xml_document * quickrank::learning::CustomLTR::get_xml_model ( ) const
virtual

Return the xml model representing the current object.

Implements quickrank::learning::LTR_Algorithm.

void quickrank::learning::CustomLTR::learn ( std::shared_ptr< data::Dataset training_dataset,
std::shared_ptr< data::Dataset validation_dataset,
std::shared_ptr< metric::ir::Metric metric,
size_t  partial_save,
const std::string  model_filename 
)
virtual

Executes the learning process.

Parameters
training_datasetThe training dataset.
validation_datasetThe validation training dataset.
metricThe metric to be optimized.
partial_saveAllows to save a partial model every given number of iterations.
model_filenameThe file where the model, and the partial models, are saved.

Implements quickrank::learning::LTR_Algorithm.

virtual std::string quickrank::learning::CustomLTR::name ( ) const
inlinevirtual

Returns the name of the ranker.

Implements quickrank::learning::LTR_Algorithm.

std::ostream & quickrank::learning::CustomLTR::put ( std::ostream &  os) const
privatevirtual

Prints the description of Algorithm, including its parameters.

Implements quickrank::learning::LTR_Algorithm.

Score quickrank::learning::CustomLTR::score_document ( const Feature d) const
virtual

Returns the score of a given document.

Implements quickrank::learning::LTR_Algorithm.

Friends And Related Function Documentation

std::ostream& operator<< ( std::ostream &  os,
const CustomLTR a 
)
friend

The output stream operator.

Member Data Documentation

const Score quickrank::learning::CustomLTR::FIXED_SCORE = 666.0
Todo:
TODO: add load_model();
const std::string quickrank::learning::CustomLTR::NAME_ = "CUSTOM"
static

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