|
DiSMEC++
|
Public Member Functions | |
| NumpyInitializationStrategy (std::shared_ptr< const WeightMatrix >, std::shared_ptr< const DenseRealVector > biases) | |
| std::unique_ptr< WeightsInitializer > | make_initializer (const std::shared_ptr< const GenericFeatureMatrix > &features) const override |
Creats a new, thread local WeightsInitializer. More... | |
Public Member Functions inherited from dismec::init::WeightInitializationStrategy | |
| virtual | ~WeightInitializationStrategy ()=default |
Private Attributes | |
| std::shared_ptr< const WeightMatrix > | m_WeightMatrix |
| std::shared_ptr< const DenseRealVector > | m_BiasVector |
|
explicit |
|
overridevirtual |
Creats a new, thread local WeightsInitializer.
This function will be called from the thread in which the returned WeightsInitializer will be used. It gets passed in a numa-local copy of the feature matrix.
| features | Read-only reference to the numa-local feature matrix. |
WeightsInitializer. Implements dismec::init::WeightInitializationStrategy.
Definition at line 54 of file numpy.cpp.
References m_BiasVector, and m_WeightMatrix.
|
private |
Definition at line 44 of file numpy.cpp.
Referenced by make_initializer().
|
private |
Definition at line 43 of file numpy.cpp.
Referenced by make_initializer().