-
Notifications
You must be signed in to change notification settings - Fork 1
Expand file tree
/
Copy pathimplementations.cpp
More file actions
28 lines (24 loc) · 852 Bytes
/
implementations.cpp
File metadata and controls
28 lines (24 loc) · 852 Bytes
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
//
// Created by Adrian Phoulady on 12/7/19.
// © 2019 Adrian Phoulady
//
#include "utils.h"
int main(int argc, char * const argv[]) {
#if defined imdb
int clauses = 3200, threshold = 12, epochs = 35;
double p = .012, gamma = .0006;
std::string const experiment = "imdb";
#elif defined connect4
int clauses = 200, threshold = 12, epochs = 200;
double p = .037, gamma = .0001;
std::string const experiment = "con4";
#else // default: mnist
int clauses = 500, threshold = 25, epochs = 400;
double p = .085, gamma = .0025;
std::string const experiment = "mnist";
#endif
bool shuffle = false, resume = false, write = false;
update(argc, argv, clauses, p, threshold, gamma, epochs, shuffle, write, resume);
fit(experiment, clauses, p, gamma, threshold, epochs, shuffle, write, resume);
return 0;
}