diff options
Diffstat (limited to 'ml_exp/read_qm7_data.py')
| -rw-r--r-- | ml_exp/read_qm7_data.py | 11 |
1 files changed, 11 insertions, 0 deletions
diff --git a/ml_exp/read_qm7_data.py b/ml_exp/read_qm7_data.py index 81be05a17..63c48195c 100644 --- a/ml_exp/read_qm7_data.py +++ b/ml_exp/read_qm7_data.py @@ -49,6 +49,17 @@ def read_nc_data(data_path): return {line[2]: int(line[0]) for line in lines} +# For use in the rewriting of the code (from functional programming to object +# oriented programming-ish) +def print_nc(nc_data): + """ + Prints the nuclear charge data to the terminal. + nc_data: dict of nc. + """ + for key in nc_data.keys(): + print(f'\'{key}\': {nc_data[key]},') + + # 'hof_qm7.txt.txt' retrieved from # https://github.com/qmlcode/tutorial def read_db_data(zi_data, |