DiSMEC++
|
Public Member Functions | |
CombinedWeightInitializer (std::unique_ptr< init::WeightsInitializer > di, std::unique_ptr< init::WeightsInitializer > si, long num_dense_features) | |
void | get_initial_weight (label_id_t label_id, Eigen::Ref< DenseRealVector > target, objective::Objective &objective) override |
Generate an initial vector for the given label. The result should be placed in target. More... | |
Public Member Functions inherited from dismec::init::WeightsInitializer | |
virtual | ~WeightsInitializer ()=default |
Public Member Functions inherited from dismec::stats::Tracked | |
Tracked () | |
Default constructor, creates the internal stats::StatisticsCollection . More... | |
void | register_stat (const std::string &name, std::unique_ptr< Statistics > stat) |
Registers a tracker for the statistics name . More... | |
std::shared_ptr< StatisticsCollection > | get_stats () const |
Gets an ownership-sharing reference to the StatisticsCollection . More... | |
Private Attributes | |
long | m_NumDenseFeatures |
std::unique_ptr< init::WeightsInitializer > | m_DenseInit |
std::unique_ptr< init::WeightsInitializer > | m_SparseInit |
Additional Inherited Members | |
Protected Member Functions inherited from dismec::stats::Tracked | |
~Tracked () | |
Non-virtual destructor. Declared protected, so we don't accidentally try to do a polymorphic delete. More... | |
template<class T > | |
void | record (stat_id_t stat, T &&value) |
Record statistics. This function just forwards all its arguments to the internal StatisticsCollection . More... | |
void | declare_stat (stat_id_t index, StatisticMetaData meta) |
Declares a new statistics. This function just forwards all its arguments to the internal StatisticsCollection . More... | |
void | declare_tag (tag_id_t index, std::string name) |
Declares a new tag. This function just forwards all its arguments to the internal StatisticsCollection . More... | |
template<class... Args> | |
void | set_tag (tag_id_t tag, long value) |
Set value of tag. This function just forwards all its arguments to the internal StatisticsCollection . More... | |
template<class... Args> | |
auto | make_timer (stat_id_t id, Args... args) |
Creates a new ScopeTimer using stats::record_scope_time . More... | |
Definition at line 19 of file cascade.cpp.
|
inline |
Definition at line 21 of file cascade.cpp.
|
inlineoverridevirtual |
Generate an initial vector for the given label. The result should be placed in target.
Implements dismec::init::WeightsInitializer.
Definition at line 26 of file cascade.cpp.
|
private |
Definition at line 33 of file cascade.cpp.
|
private |
Definition at line 32 of file cascade.cpp.
|
private |
Definition at line 34 of file cascade.cpp.