Newer
Older
import numpy as np
if __name__ == '__main__':
batch_size = np.logspace(3, 9, dtype=int, base=2, num=5)
num_epochs = np.linspace(200, 200, num=1)
gamma = 0.003
subsample_size = np.logspace(3, 9, dtype=int, base=2, num=5)
networks_types = ["nystroem", "deepstrom"]
for network in networks_types:
s_network = "--" + network + " "
s_gamma = s_network + "-G " + str(gamma) + " "
for nys_size in subsample_size:
s_nys = s_gamma + "-m " + str(nys_size) + " "
if network == "deepstrom":
for num_epoch in num_epochs:
s_epoch = s_nys + "-e" + " " + str(int(num_epoch)) + " "
for b_size in batch_size:
s_batch = s_epoch + "-s " + str(b_size) + " "
print(s_batch)
else:
print(s_nys)