DiSMEC++
dismec::init::ConstantInitializationStrategy Class Reference

An initialization strategy that sets the weight vector to a given constant. More...

Inheritance diagram for dismec::init::ConstantInitializationStrategy:
dismec::init::WeightInitializationStrategy

Public Member Functions

 ConstantInitializationStrategy (DenseRealVector vec)
 
std::unique_ptr< WeightsInitializermake_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

parallel::NUMAReplicator< DenseRealVectorm_InitVector
 

Detailed Description

An initialization strategy that sets the weight vector to a given constant.

We create NUMA-local copies of the given vector, and provide pointers to these to the resulting ConstantInitializer objects.

Definition at line 34 of file constant.cpp.

Constructor & Destructor Documentation

◆ ConstantInitializationStrategy()

ConstantInitializationStrategy::ConstantInitializationStrategy ( DenseRealVector  vec)
explicit

Definition at line 47 of file constant.cpp.

Member Function Documentation

◆ make_initializer()

std::unique_ptr< WeightsInitializer > ConstantInitializationStrategy::make_initializer ( const std::shared_ptr< const GenericFeatureMatrix > &  features) const
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.

Parameters
featuresRead-only reference to the numa-local feature matrix.
Returns
A new WeightsInitializer.

Implements dismec::init::WeightInitializationStrategy.

Definition at line 51 of file constant.cpp.

References dismec::parallel::NUMAReplicator< T >::get_local(), and m_InitVector.

Member Data Documentation

◆ m_InitVector

parallel::NUMAReplicator<DenseRealVector> dismec::init::ConstantInitializationStrategy::m_InitVector
private

Definition at line 42 of file constant.cpp.

Referenced by make_initializer().


The documentation for this class was generated from the following file: