apaam/mlpack
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Resolved Marcus' comments.

Browse Source
This commit is contained in:
Rahul Prabhu
2019-05-24 09:16:35 +05:30
parent 55814d3ca3
commit de18b388fa
2 changed files with 13 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/mlpack/methods/reinforcement_learning/environment/cart_pole.hpp
+1
View File
@@ -104,6 +104,7 @@ class CartPole
* @param tau The time interval.
* @param thetaThresholdRadians The maximum angle.
* @param xThreshold The maximum position.
* @param doneReward Reward recieved on termination.
*/
CartPole(const double gravity = 9.8,
const double massCart = 1.0,
src/mlpack/methods/reinforcement_learning/environment/multiple_pole_cart.hpp
+12 -11
View File
@@ -37,9 +37,9 @@ class MultiplePoleCart
*
* @param numPoles The number of poles.
*/
State(size_t numPoles)
State(const size_t numPoles)
{
data = arma::mat(dimension, numPoles, arma::fill::zeros);
data = arma::zeros<arma::mat>(dimension, numPoles);
}
/**
@@ -108,17 +108,18 @@ class MultiplePoleCart
* @param tau The time interval.
* @param thetaThresholdRadians The maximum angle.
* @param xThreshold The maximum position.
* @param doneReward The reward recieved on termination.
*/
MultiplePoleCart(const size_t poleNum,
const arma::vec poleLengths,
const arma::vec poleMasses,
const double gravity = 9.8,
const double massCart = 1.0,
const double forceMag = 10.0,
const double tau = 0.02,
const double thetaThresholdRadians = 12 * 2 * 3.1416 / 360,
const double xThreshold = 2.4,
const double doneReward = 0.0) :
const arma::vec& poleLengths,
const arma::vec& poleMasses,
const double gravity = 9.8,
const double massCart = 1.0,
const double forceMag = 10.0,
const double tau = 0.02,
const double thetaThresholdRadians = 12 * 2 * 3.1416 / 360,
const double xThreshold = 2.4,
const double doneReward = 0.0) :
poleNum(poleNum),
poleLengths(poleLengths),
poleMasses(poleMasses),