spec.h File Reference

#include <memory>
#include "fwd.h"
#include "matrix_types.h"
#include "spdlog/fwd.h"
#include "objective/regularizers.h"

Go to the source code of this file.

Classes
class	dismec::TrainingSpec
	This class gathers the setting-specific parts of the training process. More...
struct	dismec::DismecTrainingConfig
struct	dismec::CascadeTrainingConfig

Namespaces
	dismec
	Main namespace in which all types, classes, and functions are defined.

Typedefs
using	dismec::RegularizerSpec = std::variant< objective::SquaredNormConfig, objective::HuberConfig, objective::ElasticConfig >

Enumerations
enum class	dismec::RegularizerType {   dismec::REG_L2 , dismec::REG_L1 , dismec::REG_L1_RELAXED , dismec::REG_HUBER ,   dismec::REG_ELASTIC_50_50 , dismec::REG_ELASTIC_90_10 }
enum class	dismec::LossType { dismec::SQUARED_HINGE , dismec::LOGISTIC , dismec::HUBER_HINGE , dismec::HINGE }

Functions
std::shared_ptr< objective::Objective >	dismec::make_loss (LossType type, std::shared_ptr< const GenericFeatureMatrix > X, std::unique_ptr< objective::Objective > regularizer)
std::shared_ptr< TrainingSpec >	dismec::create_dismec_training (std::shared_ptr< const DatasetBase > data, HyperParameters params, DismecTrainingConfig config)
std::shared_ptr< TrainingSpec >	dismec::create_cascade_training (std::shared_ptr< const DatasetBase > data, std::shared_ptr< const GenericFeatureMatrix > dense, std::shared_ptr< const std::vector< std::vector< long >>> shortlist, HyperParameters params, CascadeTrainingConfig config)