summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--main.py15
1 files changed, 5 insertions, 10 deletions
diff --git a/main.py b/main.py
index d00e4157e..8bea40c96 100644
--- a/main.py
+++ b/main.py
@@ -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)