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
NNController.h
#pragma once
#include "learning/NeuralNet.h"
#include "CharController.h"
class cNNController : public cCharController
{
public:
virtual ~cNNController();
virtual int GetPoliStateSize() const;
virtual int GetPoliActionSize() const;
virtual int GetNetInputSize() const;
virtual int GetNetOutputSize() const;
virtual void RecordPoliState(Eigen::VectorXd& out_state) const;
virtual void RecordPoliAction(Eigen::VectorXd& out_action) const;
virtual double CalcReward() const;
virtual bool IsOffPolicy() const;
virtual bool LoadNet(const std::string& net_file);
virtual void LoadModel(const std::string& model_file);
virtual void LoadScale(const std::string& scale_file);
virtual void CopyNet(const cNeuralNet& net);
virtual void SaveNet(const std::string& out_file) const;
virtual void BuildNNOutputOffsetScale(Eigen::VectorXd& out_offset, Eigen::VectorXd& out_scale) const;
virtual const cNeuralNet& GetNet() const;
virtual cNeuralNet& GetNet();
protected:
cNeuralNet mNet;
cNNController();
virtual bool HasNet() const;
virtual void LoadNetIntern(const std::string& net_file);
};