-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathmain.cpp
More file actions
97 lines (68 loc) · 2.11 KB
/
main.cpp
File metadata and controls
97 lines (68 loc) · 2.11 KB
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
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
//
// Created by jamal on 30/09/2018.
//
#include <iostream>
#include "generateInput.cpp"
#include <cmath>
#include "Samplingtree.h"
#include <algorithm>
#include <cstdlib>
#include <ctime>
#include "Algorithm2.h"
#include "Algo3.h"
#include <iomanip>
int main() {
srand(time(0));
std::string dataFile, value;
long double epsilon, delta, gamma;
int n,m;
std::cout<<"Please enter input filename (or leave blank to generate input):\n";
getline(std::cin, dataFile);
std::cout<<"Epsilon:\n";
getline(std::cin, value);
epsilon = std::stod(value);
std::cout<<"Delta:\n";
getline(std::cin, value);
delta = std::stod(value);
std::cout<<"Gamma:\n";
getline(std::cin, value);
gamma = std::stod(value);
std::cout<<std::setprecision(16)<<1-gamma<<"\n";
Inputs inp;
if(!getData(dataFile, inp)){
std::cout<<"Error reading input file. Either blank or incorrect naming format. Input will be generated.\n";
std::cout<<"Number of States:\n";
getline(std::cin, value);
n = std::stoi(value);
std::cout<<"Number of Actions:\n";
getline(std::cin, value);
m = std::stoi(value);
inp = initialize(n,m,gamma);
dataFile = getCurrentTime();
dataFile = dataFile + " - data.csv";
saveData(dataFile, n, m, inp);
}else{
n = inp.N;
m = inp.M;
}
Algorithm2 *algo2 = new Algorithm2(inp, n, m);
algo2->runAlgorithmToConverge(epsilon);
std::cout<<"value iteration time: "<< algo2->getTime() <<'\n';
algo2->printV();
value_policy *vpl;
Algo3 *algo3 = new Algo3(n,m,inp);
clock_t start;
double duration;
start = clock();
vpl = algo3->HighPrecisionRandomVI(epsilon, delta);
duration = (clock() - start ) / (double) CLOCKS_PER_SEC;
std::cout<<"variance reduced time: "<< duration <<'\n';
std::cout<<"algo3:\n";
for(int i=0; i<n; i++){
std::cout<<vpl->values[i]<<" ";
}
std::cout<<"\n";
saveTimes(dataFile, n, m, inp.gamma, epsilon, delta,algo2->getTime(), duration);
clearInputs(n,m, inp);
return 0;
}