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().