-
Notifications
You must be signed in to change notification settings - Fork 1
/
param_tester.py
51 lines (33 loc) · 1.17 KB
/
param_tester.py
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
# TODO cambiar esto por el main cuando funcione
from main_tester import main
from utils import writer, remover
import os
# TODO asignar los valores que queramos para cada caso
vector_size_ls = [20]
learning_rate_ls = [0.1]
momentum_ls = [0.01]
# Second neural network parameters
feature_size_ls = [4]
epoch = 10
pooling = 'one-way pooling'
l2_penalty_ls = [0]
for vector_size in vector_size_ls:
for learning_rate in learning_rate_ls:
for momentum in momentum_ls:
for learning_rate2 in [0.1]:
for feature_size in feature_size_ls:
for l2_penalty in l2_penalty_ls:
message = f'''
########################################
The parameters we're using are the following:
vector_size = {vector_size}
learning_rate = {learning_rate}
momentum = {momentum}
learning_rate2 = {learning_rate2}
feature_size = {feature_size}
number of epochs for second neural network: {epoch}
'''
remover()
writer(message)
main(vector_size, learning_rate, momentum, learning_rate2,\
feature_size, epoch, pooling, l2_penalty)