summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--ml_exp/__main__.py51
1 files changed, 6 insertions, 45 deletions
diff --git a/ml_exp/__main__.py b/ml_exp/__main__.py
index aaeb18714..83abb918e 100644
--- a/ml_exp/__main__.py
+++ b/ml_exp/__main__.py
@@ -22,50 +22,11 @@ SOFTWARE.
"""
# from ml_exp.do_ml import do_ml
# from ml_exp.misc import plot_benchmarks
-from ml_exp.read_qm7_data import read_qm7_data
-from ml_exp.adj_matrix import fneig_matrix, adj_matrix
-from ml_exp.c_matrix import c_matrix
-from ml_exp.bob import bob
-from ml_exp.bostuff import bok_cx
+# from ml_exp.read_qm7_data import read_qm7_data
+# from ml_exp.adj_matrix import fneig_matrix, adj_matrix
+# from ml_exp.c_matrix import c_matrix
+# from ml_exp.bob import bob
+# from ml_exp.bostuff import bok_cx
if __name__ == '__main__':
- """
- do_ml(min_training_size=1500,
- max_training_size=2000,
- training_increment_size=500,
- test_size=None,
- ljm_diag_value=None,
- ljm_sigma=1.0,
- ljm_epsilon=1.0,
- r_seed=111,
- save_benchmarks=False,
- show_msgs=True)
- """
- # plot_benchmarks()
- """
- xyz, nc, pbe0, delta, atoms = read_qm7_data(return_atoms=True)
- for i in range(1):
- fnm, bonds, forces = fneig_matrix(atoms[i], xyz[i], nc[i], True)
- am = adj_matrix(bonds, forces)
-
- print(f'{i} first neighbor matrix\n{fnm}')
- print(f'{i} bond list\n{bonds}')
- print(f'{i} force list\n{forces}')
- print(f'{i} adjacency matrix\n{am}')
- print('-'*30)
- print('OK!')
- """
- xyz, nc, pbe0, delta, atoms = read_qm7_data(return_atoms=True)
- for i in range(1):
- cm = c_matrix(xyz[i], nc[i], as_eig=False)
- bob_i = bob(cm, atoms[i])
-
- print(f'{i} coulomb matrix\n{cm}')
- print(f'{i} bag of bonds\n{bob}')
-
- fnm, bonds = fneig_matrix(atoms[i], xyz[i], nc[i], False)
- am = adj_matrix(fnm, bonds, None)
- bok_cx_i = bok_cx(am, atoms[i])
-
- print(f'{i} adjacency matrix\n{am}')
- print(f'{i} bag of k_cx\n{bok_cx_i}')
+ print('Ok!')