提交 3a802729 编写于 作者: Y Yu Yang

Add comments.

上级 16ea66e8
...@@ -799,22 +799,61 @@ public: ...@@ -799,22 +799,61 @@ public:
static ParameterUpdater* createLocalUpdater(OptimizationConfig* config); static ParameterUpdater* createLocalUpdater(OptimizationConfig* config);
~ParameterUpdater(); ~ParameterUpdater();
/**
* @brief initialize Parameter Updater by GradientMachine.
* @param gm
*/
void init(const GradientMachine& gm); void init(const GradientMachine& gm);
/**
* @brief begin of a training/testing of one pass.
*/
void startPass(); void startPass();
/**
* @brief end of a traning/testing of one pass.
*/
void finishPass(); void finishPass();
/**
* @brief begin of a training/testing of one batch.
* @param data batch's size
* @return PassType, mostly will be training.
*/
PassType startBatch(size_t batchSize); PassType startBatch(size_t batchSize);
/**
* @brief end of a traning/testing of one batch
* @param cost current batch cost.
*/
void finishBatch(float cost); void finishBatch(float cost);
/**
* @brief update a parameter (by local optimizer or by cluster pserver)
* @param param
*/
void update(Parameter* param); void update(Parameter* param);
/**
* @brief restore the average parameter.
* @note It is only used in AverageOptimizer. Restore will get the current
* PARAMETER_VALUE back.
*/
void restore(); void restore();
/**
* @brief apply. Store the average parameter.
* @note It is only used in AverageOptimizer. Apply will store the current
* PARAMETER_VALUE to buffer, calcaualte current Average Parameter, and save
* it to PARAMETER_VALUE.
*/
void apply(); void apply();
/**
* @brief catchUpWith The Regularization will be delayed in many situations(
* pserver, local sparse). Catch Up means catch the regularization up, apply
* regularization to all params.
*/
void catchUpWith(); void catchUpWith();
private: private:
...@@ -830,10 +869,21 @@ private: ...@@ -830,10 +869,21 @@ private:
public: public:
~Evaluator(); ~Evaluator();
/**
* @brief begin an evaluate stage.
*/
void start(); void start();
/**
* @brief end an evaluate stage.
*/
void finish(); void finish();
/**
* @brief toString will get a evaluate result.
*
* __repr__ method in python
*/
std::string toString(); std::string toString();
private: private:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册