laPolicyStatsLogger.h

 
#ifndef LA_POLICY_STATS_LOGGER_H
#define LA_POLICY_STATS_LOGGER_H
 
#include "log/laLogger.h"
#include "tpg/policyStats.h"
#include "tpg/tpgVertex.h"
 
namespace Log {
    class LAPolicyStatsLogger : public LALogger
    {
      private:
        const TPG::TPGVertex* lastBestRoot = nullptr;
 
        uint64_t generationNumber = 0;
 
      public:
        LAPolicyStatsLogger(Learn::LearningAgent& la,
                            std::ostream& out = std::cout)
            : LALogger(la, out){};
 
        void logNewGeneration(uint64_t& generationNumber) override;
 
        void logAfterDecimate() override;
    };
}; // namespace Log
 
#endif // !LA_POLICY_STATS_LOGGER_H