dismec::init::NumpyInitializer Class Reference
Inheritance diagram for dismec::init::NumpyInitializer:
Public Member Functions | |
| NumpyInitializer (std::shared_ptr< const WeightMatrix > weights, std::shared_ptr< const DenseRealVector > biases) | |
| 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 | |
| std::shared_ptr< const WeightMatrix > | m_PreTrainedWeights |
| std::shared_ptr< const DenseRealVector > | m_Biases |
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... | |
Detailed Description
Constructor & Destructor Documentation
◆ NumpyInitializer()
|
inlineexplicit |
Member Function Documentation
◆ get_initial_weight()
|
inlineoverridevirtual |
Generate an initial vector for the given label. The result should be placed in target.
Implements dismec::init::WeightsInitializer.
Definition at line 24 of file numpy.cpp.
References m_Biases, m_PreTrainedWeights, and dismec::opaque_int_type< Tag, T >::to_index().
Member Data Documentation
◆ m_Biases
|
private |
Definition at line 35 of file numpy.cpp.
Referenced by get_initial_weight().
◆ m_PreTrainedWeights
|
private |
Definition at line 34 of file numpy.cpp.
Referenced by get_initial_weight().
The documentation for this class was generated from the following file:
- src/training/init/numpy.cpp
