Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions | Data Fields
CartPole Class Reference

Public Member Functions

torch::Tensor getState ()
 
double getReward ()
 
double isDone ()
 
void reset ()
 
void step (int action)
 

Data Fields

double gravity = 9.8
 
double masscart = 1.0
 
double masspole = 0.1
 
double total_mass = (masspole + masscart)
 
double length = 0.5
 
double polemass_length = (masspole * length)
 
double force_mag = 10.0
 
double tau = 0.02
 
double theta_threshold_radians = 12 * 2 * kPi / 360
 
double x_threshold = 2.4
 
int steps_beyond_done = -1
 
torch::Tensor state
 
double reward
 
bool done
 
int step_ = 0
 

Detailed Description

Definition at line 25 of file integration.cpp.


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