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)