https://github.com/xbpeng/DeepTerrainRL
Tip revision: ed82e2ebe5f14fa875cc3d0a2180c64980408e8f authored by Glen on 19 October 2016, 17:49:36 UTC
Update README.md
Update README.md
Tip revision: ed82e2e
ACLearner.h
#pragma once
#include "NeuralNetLearner.h"
struct cACLearner : public cNeuralNetLearner
{
public:
cACLearner(const std::shared_ptr<cNeuralNetTrainer>& trainer);
virtual ~cACLearner();
virtual void Init();
virtual void SetNet(cNeuralNet* net);
virtual const cNeuralNet* GetNet() const;
virtual void SetActorNet(cNeuralNet* net);
virtual const cNeuralNet* GetActorNet() const;
virtual void SetCriticNet(cNeuralNet* net);
virtual const cNeuralNet* GetCriticNet() const;
virtual void LoadNet(const std::string& net_file);
virtual void LoadSolver(const std::string& solver_file);
virtual void LoadActorNet(const std::string& net_file);
virtual void LoadActorSolver(const std::string& solver_file);
virtual void LoadCriticNet(const std::string& net_file);
virtual void LoadCriticSolver(const std::string& solver_file);
virtual void OutputModel(const std::string& filename) const;
virtual void OutputCritic(const std::string& filename) const;
virtual void OutputActor(const std::string& filename) const;
virtual void SyncNet();
virtual bool HasCriticNet() const;
protected:
cNeuralNet* mCriticNet;
};