Skip to content
Snippets Groups Projects
build_command_line_array_nys_vs_deepstrom.py 876 B
Newer Older
  • Learn to ignore specific revisions
  • 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)