diff options
author | David Luevano <55825613+luevano@users.noreply.github.com> | 2019-12-12 04:40:48 -0700 |
---|---|---|
committer | David Luevano <55825613+luevano@users.noreply.github.com> | 2019-12-12 04:40:48 -0700 |
commit | a62fa84e284b4c331f58d8062faa9bd74ed3f063 (patch) | |
tree | bb346f2bb77a1b48c9dbdb9e6e1fe178ad51e791 /main.py | |
parent | 0872460ee354de3d6ebdf5df229e81adfc8bdaa2 (diff) |
Reformatting
Diffstat (limited to 'main.py')
-rw-r--r-- | main.py | 15 |
1 files changed, 5 insertions, 10 deletions
@@ -34,27 +34,22 @@ from do_ml import do_ml # Initialization time. init_time = time.perf_counter() -# Move to data folder. +# Data reading. +tic = time.perf_counter() +printc('Data reading started.', 'CYAN') + init_path = os.getcwd() os.chdir('data') data_path = os.getcwd() -printc('Data reading started.', 'CYAN') -tic = time.perf_counter() - -# Read nuclear charge data. zi_data = read_nc_data(data_path) - -# Read molecule data. molecules, nuclear_charge, energy_pbe0, energy_delta = \ read_db_edata(zi_data, data_path) +os.chdir(init_path) toc = time.perf_counter() printc('\tData reading took {:.4f} seconds.'.format(toc-tic), 'GREEN') -# Go back to main folder. -os.chdir(init_path) - # Matrices calculation. cm_data = c_matrix_multiple(molecules, nuclear_charge, as_eig=True) ljm_data = lj_matrix_multiple(molecules, nuclear_charge, as_eig=True) |