TorchCraftAI
A bot for machine learning research on StarCraft: Brood War
Public Member Functions | Public Attributes | Protected Attributes | List of all members
cherrypi::CherryVisDumperModule::TreeNode Class Reference

#include <cherryvisdumper.h>

Inherits std::stringstream.

Public Member Functions

void setModule (std::string name)
 
void setFrame (FrameNum f)
 
void setId (int32_t id, std::string prefix="")
 
void addUnitWithProb (Unit *unit, float proba)
 
nlohmann::json to_json ()
 

Public Attributes

std::vector< std::shared_ptr< TreeNode > > children
 

Protected Attributes

std::unordered_map< std::string, nlohmann::json > node
 
std::vector< nlohmann::json > prob_distr
 

Member Function Documentation

void cherrypi::CherryVisDumperModule::TreeNode::addUnitWithProb ( Unit unit,
float  proba 
)
inline
void cherrypi::CherryVisDumperModule::TreeNode::setFrame ( FrameNum  f)
inline
void cherrypi::CherryVisDumperModule::TreeNode::setId ( int32_t  id,
std::string  prefix = "" 
)
inline
void cherrypi::CherryVisDumperModule::TreeNode::setModule ( std::string  name)
inline
nlohmann::json cherrypi::CherryVisDumperModule::TreeNode::to_json ( )
inline

Member Data Documentation

std::vector<std::shared_ptr<TreeNode> > cherrypi::CherryVisDumperModule::TreeNode::children
std::unordered_map<std::string, nlohmann::json> cherrypi::CherryVisDumperModule::TreeNode::node
protected
std::vector<nlohmann::json> cherrypi::CherryVisDumperModule::TreeNode::prob_distr
protected

The documentation for this class was generated from the following file: