25 #ifndef __IMPLICIT_ALS_TRAINING_INIT_DISTRIBUTED_H__
26 #define __IMPLICIT_ALS_TRAINING_INIT_DISTRIBUTED_H__
28 #include "algorithms/algorithm.h"
29 #include "algorithms/implicit_als/implicit_als_training_init_types.h"
35 namespace implicit_als
54 template<ComputeStep step,
typename algorithmFPType, Method method, CpuType cpu>
55 class DistributedContainer {};
61 template<
typename algorithmFPType, Method method, CpuType cpu>
62 class DistributedContainer<step1Local, algorithmFPType, method, cpu> :
public
63 TrainingContainerIface<distributed>
71 DistributedContainer(daal::services::Environment::env *daalEnv);
73 ~DistributedContainer();
78 services::Status compute() DAAL_C11_OVERRIDE;
83 services::Status finalizeCompute() DAAL_C11_OVERRIDE {
return services::Status(); }
90 template<
typename algorithmFPType, Method method, CpuType cpu>
91 class DistributedContainer<step2Local, algorithmFPType, method, cpu> :
public
92 TrainingContainerIface<distributed>
100 DistributedContainer(daal::services::Environment::env *daalEnv);
102 ~DistributedContainer();
107 services::Status compute() DAAL_C11_OVERRIDE;
112 services::Status finalizeCompute() DAAL_C11_OVERRIDE {
return services::Status(); }
128 template<ComputeStep step,
typename algorithmFPType = DAAL_ALGORITHM_FP_TYPE, Method method = fastCSR>
129 class DAAL_EXPORT Distributed {};
142 template<
typename algorithmFPType, Method method>
143 class DAAL_EXPORT Distributed<step1Local, algorithmFPType, method> :
public Training<distributed>
146 typedef algorithms::implicit_als::training::init::DistributedInput<step1Local> InputType;
147 typedef algorithms::implicit_als::training::init::DistributedParameter ParameterType;
148 typedef algorithms::implicit_als::training::init::Result ResultType;
149 typedef algorithms::implicit_als::training::init::PartialResult PartialResultType;
151 DistributedInput<step1Local> input;
152 ParameterType parameter;
166 Distributed(
const Distributed<step1Local, algorithmFPType, method> &other) : input(other.input), parameter(other.parameter)
175 virtual int getMethod() const DAAL_C11_OVERRIDE {
return(
int)method; }
181 services::Status setPartialResult(
const PartialResultPtr& partialResult)
183 DAAL_CHECK(partialResult, services::ErrorNullPartialResult);
184 _partialResult = partialResult;
185 _pres = _partialResult.get();
186 return services::Status();
193 PartialResultPtr getPartialResult() {
return _partialResult; }
200 services::SharedPtr<Distributed<step1Local, algorithmFPType, method> > clone()
const
202 return services::SharedPtr<Distributed<step1Local, algorithmFPType, method> >(cloneImpl());
206 PartialResultPtr _partialResult;
208 virtual Distributed<step1Local, algorithmFPType, method> * cloneImpl() const DAAL_C11_OVERRIDE
210 return new Distributed<step1Local, algorithmFPType, method>(*this);
213 services::Status allocateResult() DAAL_C11_OVERRIDE
215 return services::Status();
218 services::Status allocatePartialResult() DAAL_C11_OVERRIDE
220 services::Status s = _partialResult->allocate<algorithmFPType>(&input, ¶meter, method);
221 _pres = _partialResult.get();
225 services::Status initializePartialResult() DAAL_C11_OVERRIDE
227 return services::Status();
232 _ac =
new __DAAL_ALGORITHM_CONTAINER(distributed, DistributedContainer, step1Local, algorithmFPType, method)(&_env);
235 _partialResult.reset(
new PartialResultType());
250 template<
typename algorithmFPType, Method method>
251 class DAAL_EXPORT Distributed<step2Local, algorithmFPType, method> :
public Training<distributed>
254 typedef algorithms::implicit_als::training::init::DistributedInput<step2Local> InputType;
255 typedef algorithms::implicit_als::training::init::Result ResultType;
256 typedef algorithms::implicit_als::training::init::DistributedPartialResultStep2 PartialResultType;
258 DistributedInput<step2Local> input;
272 Distributed(
const Distributed<step2Local, algorithmFPType, method> &other)
275 input.set(inputOfStep2FromStep1, other.input.get(inputOfStep2FromStep1));
282 virtual int getMethod() const DAAL_C11_OVERRIDE {
return(
int)method; }
288 services::Status setPartialResult(
const DistributedPartialResultStep2Ptr& partialResult)
290 DAAL_CHECK(partialResult, services::ErrorNullPartialResult);
291 _partialResult = partialResult;
292 _pres = _partialResult.get();
293 return services::Status();
300 DistributedPartialResultStep2Ptr getPartialResult() {
return _partialResult; }
307 services::SharedPtr<Distributed<step2Local, algorithmFPType, method> > clone()
const
309 return services::SharedPtr<Distributed<step2Local, algorithmFPType, method> >(cloneImpl());
313 DistributedPartialResultStep2Ptr _partialResult;
315 virtual Distributed<step2Local, algorithmFPType, method> * cloneImpl() const DAAL_C11_OVERRIDE
317 return new Distributed<step2Local, algorithmFPType, method>(*this);
320 services::Status allocateResult() DAAL_C11_OVERRIDE
322 return services::Status();
325 services::Status allocatePartialResult() DAAL_C11_OVERRIDE
327 services::Status s = _partialResult->allocate<algorithmFPType>(&input, NULL, method);
328 _pres = _partialResult.get();
332 services::Status initializePartialResult() DAAL_C11_OVERRIDE
334 return services::Status();
339 _ac =
new __DAAL_ALGORITHM_CONTAINER(distributed, DistributedContainer, step2Local, algorithmFPType, method)(&_env);
342 _partialResult.reset(
new PartialResultType());
347 using interface1::DistributedContainer;
348 using interface1::Distributed;
daal::step1Local
Definition: daal_defines.h:123
daal::algorithms::implicit_als::training::init::interface1::Distributed< step2Local, algorithmFPType, method >::setPartialResult
services::Status setPartialResult(const DistributedPartialResultStep2Ptr &partialResult)
Definition: implicit_als_training_init_distributed.h:288
daal::algorithms::implicit_als::training::init::inputOfStep2FromStep1
Definition: implicit_als_training_init_types.h:79
daal::algorithms::implicit_als::training::init::interface1::Distributed< step1Local, algorithmFPType, method >::Distributed
Distributed(const Distributed< step1Local, algorithmFPType, method > &other)
Definition: implicit_als_training_init_distributed.h:166
daal::algorithms::implicit_als::training::init::interface1::Distributed< step2Local, algorithmFPType, method >::Distributed
Distributed()
Definition: implicit_als_training_init_distributed.h:261
daal::algorithms::implicit_als::training::init::interface1::Distributed< step1Local, algorithmFPType, method >::setPartialResult
services::Status setPartialResult(const PartialResultPtr &partialResult)
Definition: implicit_als_training_init_distributed.h:181
daal::algorithms::implicit_als::training::init::interface1::DistributedInput< step2Local >
Input objects for the implicit ALS initialization algorithm in the second step of the distributed pro...
Definition: implicit_als_training_init_types.h:275
daal::algorithms::implicit_als::training::init::interface1::DistributedInput< step2Local >::get
data_management::KeyValueDataCollectionPtr get(Step2LocalInputId id) const
daal::algorithms::implicit_als::training::init::interface1::Distributed< step2Local, algorithmFPType, method >::getMethod
virtual int getMethod() const DAAL_C11_OVERRIDE
Definition: implicit_als_training_init_distributed.h:282
daal::algorithms::implicit_als::training::init::interface1::Distributed< step1Local, algorithmFPType, method >::Distributed
Distributed()
Definition: implicit_als_training_init_distributed.h:155
daal::algorithms::implicit_als::training::init::interface1::Distributed< step1Local, algorithmFPType, method >::getMethod
virtual int getMethod() const DAAL_C11_OVERRIDE
Definition: implicit_als_training_init_distributed.h:175
daal::algorithms::implicit_als::training::init::interface1::Distributed< step2Local, algorithmFPType, method >::input
DistributedInput< step2Local > input
Definition: implicit_als_training_init_distributed.h:258
daal::algorithms::implicit_als::training::init::interface1::Distributed< step1Local, algorithmFPType, method >::getPartialResult
PartialResultPtr getPartialResult()
Definition: implicit_als_training_init_distributed.h:193
daal::algorithms::implicit_als::training::init::interface1::Distributed< step2Local, algorithmFPType, method >
Initializes the implicit ALS model in the second step of the distributed processing mode...
Definition: implicit_als_training_init_distributed.h:251
daal::algorithms::implicit_als::training::init::interface1::Distributed< step1Local, algorithmFPType, method >::input
DistributedInput< step1Local > input
Definition: implicit_als_training_init_distributed.h:151
daal::algorithms::kmeans::init::interface2::Distributed
class DAAL_EXPORT Distributed
Computes initial clusters for K-Means algorithm in the distributed processing mode.
Definition: kmeans_init_distributed.h:277
daal::algorithms::implicit_als::training::init::interface1::Distributed< step1Local, algorithmFPType, method >::clone
services::SharedPtr< Distributed< step1Local, algorithmFPType, method > > clone() const
Definition: implicit_als_training_init_distributed.h:200
daal::algorithms::Training
Provides methods to train models that depend on the data provided. For example, these methods enable ...
Definition: training.h:62
daal::algorithms::implicit_als::training::init::interface1::DistributedContainer< step2Local, algorithmFPType, method, cpu >::finalizeCompute
services::Status finalizeCompute() DAAL_C11_OVERRIDE
Definition: implicit_als_training_init_distributed.h:112
daal::algorithms::implicit_als::training::init::interface1::Distributed
Initializes the implicit ALS model in the distributed processing mode.
Definition: implicit_als_training_init_distributed.h:129
daal::algorithms::implicit_als::training::init::interface1::Distributed< step1Local, algorithmFPType, method >
Initializes the implicit ALS model in the first step of the distributed processing mode...
Definition: implicit_als_training_init_distributed.h:143
daal::algorithms::implicit_als::training::init::interface1::Distributed< step2Local, algorithmFPType, method >::getPartialResult
DistributedPartialResultStep2Ptr getPartialResult()
Definition: implicit_als_training_init_distributed.h:300
daal::algorithms::implicit_als::training::init::interface1::DistributedInput< step1Local >
Input objects for the implicit ALS initialization algorithm in the first step of the distributed proc...
Definition: implicit_als_training_init_types.h:254
daal::algorithms::implicit_als::training::init::interface1::Distributed< step1Local, algorithmFPType, method >::parameter
ParameterType parameter
Definition: implicit_als_training_init_distributed.h:152
daal::algorithms::implicit_als::training::init::interface1::DistributedContainer
Class containing methods to compute the results of the implicit ALS initialization algorithm in the d...
Definition: implicit_als_training_init_distributed.h:55
daal::services::ErrorNullPartialResult
Definition: error_indexes.h:107
daal::distributed
Definition: daal_defines.h:113
daal::algorithms::implicit_als::training::init::interface1::Distributed< step2Local, algorithmFPType, method >::clone
services::SharedPtr< Distributed< step2Local, algorithmFPType, method > > clone() const
Definition: implicit_als_training_init_distributed.h:307
daal::algorithms::TrainingContainerIface
Abstract interface class that provides virtual methods to access and run implementations of the model...
Definition: training.h:52
daal::step2Local
Definition: daal_defines.h:128
daal::algorithms::implicit_als::training::init::interface1::Distributed< step2Local, algorithmFPType, method >::Distributed
Distributed(const Distributed< step2Local, algorithmFPType, method > &other)
Definition: implicit_als_training_init_distributed.h:272
daal::algorithms::implicit_als::training::init::interface1::DistributedContainer< step1Local, algorithmFPType, method, cpu >::finalizeCompute
services::Status finalizeCompute() DAAL_C11_OVERRIDE
Definition: implicit_als_training_init_distributed.h:83