-
Notifications
You must be signed in to change notification settings - Fork 0
/
Mdp.java
28 lines (26 loc) · 915 Bytes
/
Mdp.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
import java.util.List;
import java.util.Map;
public class Mdp {
Graph g;
double[][] transitionMatrix;
double[] rewards;
double discountFactor;
int iterationCount;
double tolerance;
boolean minimizeCost;
double[] values;
List<String> decisionNodeNames;
Map<String, String> policy;
public Mdp(Graph g, double[][] transitionMatrix, double[] rewards, double discountFactor, int iterationCount, double tolerance, boolean minimizeCost) {
this.g = g;
this.transitionMatrix = transitionMatrix;
this.rewards = rewards;
this.discountFactor = discountFactor;
this.iterationCount = iterationCount;
this.tolerance = tolerance;
this.minimizeCost = minimizeCost;
this.values = Solver.getRewards(g);
this.decisionNodeNames = Solver.getDecisionNodeNames(g);
this.policy = Solver.getIntialPolicy();
}
}