Add adjacency matrix

1 files changed, 14 insertions, 1 deletions

diff --git a/ml_exp/__main__.py b/ml_exp/__main__.py
index 167f0fa7b..63708e75c 100644
--- a/ml_exp/__main__.py
+++ b/ml_exp/__main__.py
@@ -20,10 +20,13 @@ LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 SOFTWARE.
 """
-from ml_exp.do_ml import do_ml
+# 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
 
 if __name__ == '__main__':
+    """
     do_ml(min_training_size=1500,
           max_training_size=2000,
           training_increment_size=500,
@@ -34,5 +37,15 @@ if __name__ == '__main__':
           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 = fneig_matrix(atoms[i], xyz[i])
+        am = adj_matrix(bonds)
+
+        print(f'{i} first neighbor matrix\n{fnm}')
+        print(f'{i} bond list\n{bonds}')
+        print(f'{i} adjacency matrix\n{am}')
+        print('-'*30)
     print('OK!')