File head_type_partial.hpp

namespace seco
class PartialHeadConfig : public seco::IHeadConfig
#include <head_type_partial.hpp>

Allows to configure partial rule heads that predict for a subset of the available labels.

Public Functions

PartialHeadConfig(const std::unique_ptr<IHeuristicConfig> &heuristicConfigPtr, const std::unique_ptr<IHeuristicConfig> &pruningHeuristicConfigPtr, const std::unique_ptr<ILiftFunctionConfig> &liftFunctionConfigPtr)
Parameters:

  • heuristicConfigPtr – A reference to an unique pointer that stores the configuration of the heuristic for learning rules

  • pruningHeuristicConfigPtr – A reference to an unique pointer that stores the configuration of the heuristic for pruning rules

  • liftFunctionConfigPtr – A reference to an unique pointer that stores the configuration of the lift function that should affect the quality of rules, depending on the number of labels for which they predict

virtual std::unique_ptr<IStatisticsProviderFactory> createStatisticsProviderFactory(const IRowWiseLabelMatrix &labelMatrix) const override

Creates and returns a new object of type IStatisticsProviderFactory according to the specified configuration.

Parameters:

labelMatrix – A reference to an object of type IRowWiseLabelMatrix that provides row-wise access to the labels of the training examples

Returns:

An unique pointer to an object of type IStatisticsProviderFactory that has been created

Private Members

const std::unique_ptr<IHeuristicConfig> &heuristicConfigPtr_
const std::unique_ptr<IHeuristicConfig> &pruningHeuristicConfigPtr_
const std::unique_ptr<ILiftFunctionConfig> &liftFunctionConfigPtr_