|
|
@ -61,9 +61,6 @@ public class MDPRewardsSimple implements MDPRewards |
|
|
*/ |
|
|
*/ |
|
|
public void setStateReward(int s, double r) |
|
|
public void setStateReward(int s, double r) |
|
|
{ |
|
|
{ |
|
|
// Nothing to do for zero reward |
|
|
|
|
|
if (r == 0.0) |
|
|
|
|
|
return; |
|
|
|
|
|
// If no rewards array created yet, create it |
|
|
// If no rewards array created yet, create it |
|
|
if (stateRewards == null) { |
|
|
if (stateRewards == null) { |
|
|
stateRewards = new ArrayList<Double>(numStates); |
|
|
stateRewards = new ArrayList<Double>(numStates); |
|
|
@ -80,9 +77,6 @@ public class MDPRewardsSimple implements MDPRewards |
|
|
public void setTransitionReward(int s, int i, double r) |
|
|
public void setTransitionReward(int s, int i, double r) |
|
|
{ |
|
|
{ |
|
|
List<Double> list; |
|
|
List<Double> list; |
|
|
// Nothing to do for zero reward |
|
|
|
|
|
if (r == 0.0) |
|
|
|
|
|
return; |
|
|
|
|
|
// If no rewards array created yet, create it |
|
|
// If no rewards array created yet, create it |
|
|
if (transRewards == null) { |
|
|
if (transRewards == null) { |
|
|
transRewards = new ArrayList<List<Double>>(numStates); |
|
|
transRewards = new ArrayList<List<Double>>(numStates); |
|
|
|